Searched refs:inv_stddev (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 62 se::DeviceMemory<float> inv_stddev; member
199 params->inv_stddev, // in RunCudnnBatchNormBackwardImpl()
289 se::DeviceMemory<float> mean, se::DeviceMemory<float> inv_stddev, in RunCudnnBatchNormBackward() argument
298 backward_params.inv_stddev = inv_stddev; in RunCudnnBatchNormBackward()
|
| D | cudnn_batchnorm_thunk.cc | 121 const BufferAllocation::Slice& inv_stddev, in CudnnBatchNormBackwardThunk() argument
131 inv_stddev_(inv_stddev), in CudnnBatchNormBackwardThunk()
|
| D | cudnn_batchnorm_runner.h | 59 se::DeviceMemory<float> mean, se::DeviceMemory<float> inv_stddev,
|
| D | cudnn_batchnorm_thunk.h | 110 const BufferAllocation::Slice& inv_stddev,
|
| D | ir_emitter_unnested.cc | 1409 TF_ASSIGN_OR_RETURN(auto inv_stddev, in EmitBatchNormThunkFromMlir()
1435 /*inv_stddev=*/inv_stddev, in EmitBatchNormThunkFromMlir()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_instance_norm.cc | 154 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()
|