| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 60 Tensor y_backprop; member
147 Output y_backprop = ops::Const(root.WithOpName("y_backprop"), in RunFusedBatchNorm() local
238 Output y_backprop = ops::Const(root.WithOpName("y_backprop"), in RunFusedBatchNormEx() local
389 Tensor y_backprop(t_dtype, input_shape); in VerifyTensorsNear() local
465 [&](const Tensor& y_backprop, const Tensor& input_data, in VerifyFusedBatchNormEx()
469 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
|
| D | fused_batch_norm_op_test.cc | 278 Node* y_backprop = test::graph::Constant(g, y_backprop_t, "y_backprop"); in FusedBatchNormGrad() local
|
| D | fused_batch_norm_op.cc | 406 typename TTypes<T, 4>::Tensor y_backprop( in operator ()() local
538 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
1492 Tensor y_backprop = context->input(0); in ComputeWithReservedSpace() local
|
| D | fused_batch_norm_op.cu.cc | 43 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 148 Tensor y_backprop(dtype, {batch, height, width, out_channels}); in VerifyTensorsCloseForGrad() local
356 const float epsilon) { in VerifyFusedBatchNormGradWithConv2D()
417 const float epsilon) { in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1269 ShapeHandle y_backprop; in FusedBatchNormGradShape() local
1318 ShapeHandle y_backprop; in FusedBatchNormGradExShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.h | 1217 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
1235 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
|
| D | stream.cc | 356 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x, in ThenBatchNormalizationBackward()
410 const DeviceMemory<Eigen::half> &y_backprop, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3727 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
3745 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
3765 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 256 auto y_backprop = in CreateSimpleFusedBatchNormGrad() local
699 auto* y_backprop = context.graph_view->GetNode("y_backprop"); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 5605 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
5626 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
5649 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl()
|