| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op.cc | 412 typename TTypes<T, 4>::Tensor x_backprop( in operator ()() local
544 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator ()() local
1574 Tensor* x_backprop = nullptr; in ComputeWithReservedSpace() local
|
| H A D | fused_batch_norm_op.cu.cc | 49 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator ()() local
|
| H A D | fused_batch_norm_ex_op_test.cc | 61 Tensor x_backprop; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | batch_norm_op.cc | 275 xla::XlaOp x_backprop; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op_test.cc | 386 auto x_backprop = in VerifyFusedBatchNormGradWithConv2D() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer_test.cc | 274 auto x_backprop = ops::Identity(scope.WithOpName("x_backprop"), in CreateSimpleFusedBatchNormGrad() local
1202 auto* x_backprop = context.graph_view->GetNode("x_backprop"); in TEST_F() local
1265 auto* x_backprop = context.graph_view->GetNode("x_backprop"); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1223 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
1242 DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward()
|
| H A D | stream.cc | 361 dnn::ActivationMode activation_mode, DeviceMemory<float> *x_backprop, in ThenBatchNormalizationBackward()
416 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *x_backprop, in ThenBatchNormalizationBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3733 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward()
3751 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
3769 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 5611 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
5632 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward()
5654 dnn::ActivationMode activation_mode, DeviceMemory<T>* x_backprop, in DoBatchNormalizationBackwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 1289 ShapeHandle x_backprop; in FusedBatchNormGradShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 2239 Value x_backprop, scale_backprop, offset_backprop; in matchAndRewrite() local
|