Searched defs:inv_var (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 50 T inv_var = variance[index]; in InvVarianceToVarianceKernel() local
|
| D | fused_batch_norm_op.cc | 333 Tensor inv_var; in operator ()() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3282 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
3297 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
3314 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
420 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3087 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward()
|