Searched defs:inv_var (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1067 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
1083 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
|
| D | stream.cc | 373 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
424 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3645 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
3663 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
3683 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3613 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
|