| /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
239 Output y_backprop = ops::Const(root.WithOpName("y_backprop"), in RunFusedBatchNormEx() local
368 Tensor y_backprop(t_dtype, input_shape); in VerifyTensorsNear() local
444 [&](const Tensor& y_backprop, const Tensor& input_data, in VerifyFusedBatchNormEx()
448 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
|
| D | fused_batch_norm_op_test.cc | 257 Node* y_backprop = test::graph::Constant(g, y_backprop_t, "y_backprop"); in FusedBatchNormGrad() local
|
| D | fused_batch_norm_op.cc | 385 typename TTypes<T, 4>::Tensor y_backprop( in operator ()() local
523 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
1399 Tensor y_backprop = context->input(0); in ComputeWithReservedSpace() local
|
| D | fused_batch_norm_op.cu.cc | 42 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 147 Tensor y_backprop(dtype, {batch, height, width, out_channels}); in VerifyTensorsCloseForGrad() local
353 const float epsilon) { in VerifyFusedBatchNormGradWithConv2D()
414 const float epsilon) { in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1065 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
1081 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
|
| D | stream.cc | 371 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x, in ThenBatchNormalizationBackward()
422 const DeviceMemory<Eigen::half> &y_backprop, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3643 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
3661 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
3681 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
|
| D | layout_optimizer_test.cc | 205 Output y_backprop = in SimpleFusedBatchNormGrad() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3611 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward()
3627 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward()
3645 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1214 ShapeHandle y_backprop; in FusedBatchNormGradShape() local
|