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.cc5262 float batchnorm_data[4]; in ConvertFusedBatchNorm() local
5266 batchnorm_data[j] = vals_array[j][i]; in ConvertFusedBatchNorm()
5268 batchnorm_data[j] = in ConvertFusedBatchNorm()
5273 batchnorm_data[j] = vals_array[j][0]; in ConvertFusedBatchNorm()
5275 batchnorm_data[j] = in ConvertFusedBatchNorm()
5280 float scale = batchnorm_data[0]; in ConvertFusedBatchNorm()
5281 float offset = batchnorm_data[1]; in ConvertFusedBatchNorm()
5282 float mean = batchnorm_data[2]; in ConvertFusedBatchNorm()
5283 float variance = batchnorm_data[3]; in ConvertFusedBatchNorm()