Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc62 se::DeviceMemory<float> inv_stddev; member
289 se::DeviceMemory<float> mean, se::DeviceMemory<float> inv_stddev, in RunCudnnBatchNormBackward()
Dcudnn_batchnorm_thunk.cc121 const BufferAllocation::Slice& inv_stddev, in CudnnBatchNormBackwardThunk()
/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc154 const float32x4_t inv_stddev[4] = { in MinAndMax() local
211 const float32x4_t inv_stddev[4] = { in InstanceNorm() local