Home
last modified time | relevance | path

Searched defs:scale_backprop (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc146 xla::XlaOp scale_backprop; in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc171 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
650 Tensor* scale_backprop = nullptr; in Compute() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3285 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3300 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3316 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3090 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3104 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
3119 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc372 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
423 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()