Home
last modified time | relevance | path

Searched refs:inv_var (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h249 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
260 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
608 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc,
Dcuda_dnn.cc3645 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument
3654 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
3663 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument
3672 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
3683 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl() argument
3735 /*savedInvVariance=*/inv_var.opaque(), in DoBatchNormalizationBackwardImpl()
3750 mean.opaque(), inv_var.opaque())); in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1067 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument
1083 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument
Dstream.h264 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
289 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var,
Dstream.cc373 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward() argument
384 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
424 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var, in ThenBatchNormalizationBackward() argument
436 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h298 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
Drocm_dnn.cc3613 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument
3621 stream, miopenHalf, miopenFloat, y_backprop, x, scale, mean, inv_var, in DoBatchNormalizationBackward()