Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc154 const float32x4_t inv_stddev[4] = { in MinAndMax() local
176 vmulq_f32(vsubq_f32(v_float[i], mean[i]), inv_stddev[i]); in MinAndMax()
211 const float32x4_t inv_stddev[4] = { in InstanceNorm() local
233 vmulq_f32(vsubq_f32(v_float[i], mean[i]), inv_stddev[i]); in InstanceNorm()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc217 const BufferAllocation::Slice& inv_stddev, in CudnnBatchNormBackwardThunk() argument
227 inv_stddev_(inv_stddev), in CudnnBatchNormBackwardThunk()
Dcudnn_batchnorm_thunk.h115 const BufferAllocation::Slice& inv_stddev,