Searched defs:estimated_variance (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op.cc | 299 typename TTypes<U>::ConstVec estimated_variance( in operator ()() local
1274 const Tensor& estimated_variance = context->input(4); in ComputeWithReservedSpace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1166 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
1185 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
|
| H A D | stream.cc | 332 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward()
385 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3640 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3661 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3683 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 5412 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
5435 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
5461 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl()
|