Home
last modified time | relevance | path

Searched defs:side_input_backprop (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1225 DeviceMemory<float>* side_input_backprop, in DoBatchNormalizationBackward()
1244 DeviceMemory<Eigen::half>* side_input_backprop, in DoBatchNormalizationBackward()
H A Dstream.cc363 DeviceMemory<float> *side_input_backprop, in ThenBatchNormalizationBackward()
418 DeviceMemory<Eigen::half> *side_input_backprop, in ThenBatchNormalizationBackward()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc1596 Tensor* side_input_backprop = nullptr; in ComputeWithReservedSpace() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc5613 DeviceMemory<float>* side_input_backprop, in DoBatchNormalizationBackward()
5634 DeviceMemory<Eigen::half>* side_input_backprop, in DoBatchNormalizationBackward()
5656 DeviceMemory<T>* side_input_backprop, in DoBatchNormalizationBackwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc1328 ShapeHandle side_input_backprop; in FusedBatchNormGradExShape() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc3735 DeviceMemory<Eigen::half>* side_input_backprop, in DoBatchNormalizationBackward()
3753 DeviceMemory<float>* side_input_backprop, in DoBatchNormalizationBackward()