Home
last modified time | relevance | path

Searched defs:estimated_variance (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc82 typename TTypes<U>::ConstVec estimated_variance( in operator ()() local
542 const Tensor& estimated_variance = context->input(4); in Compute() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3182 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3203 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3226 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3000 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3019 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward()
3040 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc343 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward()
393 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward()