Home
last modified time | relevance | path

Searched refs:batchnorm_data (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3666 float batchnorm_data[4]; in ConvertFusedBatchNorm() local
3670 batchnorm_data[j] = vals_array[j][i]; in ConvertFusedBatchNorm()
3672 batchnorm_data[j] = in ConvertFusedBatchNorm()
3677 batchnorm_data[j] = vals_array[j][0]; in ConvertFusedBatchNorm()
3679 batchnorm_data[j] = in ConvertFusedBatchNorm()
3684 float scale = batchnorm_data[0]; in ConvertFusedBatchNorm()
3685 float offset = batchnorm_data[1]; in ConvertFusedBatchNorm()
3686 float mean = batchnorm_data[2]; in ConvertFusedBatchNorm()
3687 float variance = batchnorm_data[3]; in ConvertFusedBatchNorm()