Home
last modified time | relevance | path

Searched defs:estimated_mean (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc286 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()() local
1247 const Tensor& estimated_mean = context->input(3); in ComputeWithReservedSpace() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1014 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
1033 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
Dstream.cc345 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward()
396 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3450 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3473 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3498 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3524 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3544 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward()
3566 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl()