models.neural_network.optimization_utils

Neural Network optimization utilities.

Module Contents

Functions

_fuse_layer_with_scale_layer(layer_idx,scale_idx,layers)
_fuse_layer_with_bias_layer(layer_idx,bias_idx,layers)
_bn_scale_fusion(bn_idx,scale_idx,layers)
_conv_bn_fusion(conv_idx,bn_idx,layers)
_get_nn_mappings(layers)
_optimize_nn(layers)
_fuse_layer_with_scale_layer(layer_idx, scale_idx, layers)
_fuse_layer_with_bias_layer(layer_idx, bias_idx, layers)
_bn_scale_fusion(bn_idx, scale_idx, layers)
_conv_bn_fusion(conv_idx, bn_idx, layers)
_get_nn_mappings(layers)
_optimize_nn(layers)