Searched defs:x_backprop (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 145 xla::XlaOp x_backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 170 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator ()() local
645 Tensor* x_backprop = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 2478 ShapeHandle x_backprop; in __anon3e672dd83f02() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 524 auto x_backprop = SimpleFusedBatchNormGrad(&s, true); in TEST_F() local
538 auto x_backprop = SimpleFusedBatchNormGrad(&s, false); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3285 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/core/framework/ |
| D | common_shape_fns.cc | 851 ShapeHandle x_backprop; in FusedBatchNormGradShape() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3090 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/ |
| D | stream.cc | 372 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
423 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward()
|