Searched defs:scale_backprop (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 147 xla::ComputationDataHandle scale_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 150 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
629 Tensor* scale_backprop = nullptr; in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 450 Tensor* scale_backprop = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2856 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
2870 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward()
2885 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 347 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
398 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
|