Searched defs:estimated_mean (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 298 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()() local
1273 const Tensor& estimated_mean = context->input(3); in ComputeWithReservedSpace() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.h | 1165 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
1184 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
|
| D | stream.cc | 331 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward()
384 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3639 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3660 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3682 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 5411 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
5434 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
5460 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl()
|