| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 60 Tensor y_backprop; member
69 const Tensor& y_backprop, const Tensor& input_data,
147 Output y_backprop = ops::Const(root.WithOpName("y_backprop"), in RunFusedBatchNorm() local
189 root.WithOpName("activation_grad"), y_backprop, activation); in RunFusedBatchNorm()
192 ops::Identity(root.WithOpName("activation_grad"), y_backprop); in RunFusedBatchNorm()
218 backward->y_backprop = out_tensors[6]; in RunFusedBatchNorm()
239 Output y_backprop = ops::Const(root.WithOpName("y_backprop"), in RunFusedBatchNormEx() local
285 .Input({y_backprop.name(), 0, t_dtype}) in RunFusedBatchNormEx()
291 .Input({y_backprop.name(), 0, t_dtype}) in RunFusedBatchNormEx()
330 backward->y_backprop = out_tensors[6]; in RunFusedBatchNormEx()
[all …]
|
| D | fused_batch_norm_op.cc | 385 typename TTypes<T, 4>::Tensor y_backprop( in operator ()() local
465 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator ()()
523 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
585 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator ()()
985 void operator()(OpKernelContext* context, const Tensor& y_backprop, in operator ()()
1019 << " y_backprop shape: " << y_backprop.shape().DebugString() in operator ()()
1026 Tensor y_backprop_maybe_transformed = y_backprop; in operator ()()
1399 Tensor y_backprop = context->input(0); in ComputeWithReservedSpace() local
1411 OP_REQUIRES(context, y_backprop.dims() == 4 || y_backprop.dims() == 5, in ComputeWithReservedSpace()
1413 y_backprop.shape().DebugString())); in ComputeWithReservedSpace()
[all …]
|
| D | fused_batch_norm_op_test.cc | 257 Node* y_backprop = test::graph::Constant(g, y_backprop_t, "y_backprop"); in FusedBatchNormGrad() local
264 .Input(y_backprop) in FusedBatchNormGrad()
|
| D | fused_batch_norm_op.cu.cc | 42 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
85 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 51 const Tensor& input, const Tensor& filter, const Tensor& y_backprop,
147 Tensor y_backprop(dtype, {batch, height, width, out_channels}); in VerifyTensorsCloseForGrad() local
148 y_backprop.flat<T>() = in VerifyTensorsCloseForGrad()
149 y_backprop.flat<T>().template setRandom<random_gen_>(); in VerifyTensorsCloseForGrad()
168 run(input, filter, y_backprop, scale, mean, variance, res_sp3, &output, in VerifyTensorsCloseForGrad()
171 run_mkl(input, filter, y_backprop, scale, mean, variance, res_sp3, in VerifyTensorsCloseForGrad()
349 const Tensor& y_backprop, const Tensor& scale, in VerifyFusedBatchNormGradWithConv2D()
366 Input::Initializer(y_backprop)); in VerifyFusedBatchNormGradWithConv2D()
410 const Tensor& y_backprop, const Tensor& scale, in VerifyFusedBatchNormGradWithConv2D()
442 AddInputFromArray<T>(y_backprop.shape(), y_backprop.flat<T>()); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormGradV3.pbtxt | 4 name: "y_backprop"
98 name: "y_backprop"
|
| D | FusedBatchNormGrad.pbtxt | 4 name: "y_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 4 name: "y_backprop"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 4 name: "y_backprop"
86 The data format for y_backprop, x, x_backprop.
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 4 name: "y_backprop"
92 The data format for y_backprop, x, x_backprop.
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 4 name: "y_backprop"
100 The data format for y_backprop, x, x_backprop.
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormGrad.pbtxt | 4 name: "y_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 4 name: "y_backprop"
|
| D | FusedBatchNormGradV3.pbtxt | 4 name: "y_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 247 Stream* stream, const DeviceMemory<float>& y_backprop,
258 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop,
606 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x,
|
| D | cuda_dnn.cc | 3643 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward() argument
3653 stream, CUDNN_DATA_FLOAT, CUDNN_DATA_FLOAT, y_backprop, in DoBatchNormalizationBackward()
3661 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward() argument
3671 stream, CUDNN_DATA_HALF, CUDNN_DATA_FLOAT, y_backprop, in DoBatchNormalizationBackward()
3681 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl() argument
3723 /*dyData=*/y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
3746 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 285 Stream* stream, const DeviceMemory<float>& y_backprop,
296 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop,
721 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x,
|
| D | rocm_dnn.cc | 3611 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward() argument
3621 stream, miopenHalf, miopenFloat, y_backprop, x, scale, mean, inv_var, in DoBatchNormalizationBackward()
3627 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward() argument
3637 stream, miopenFloat, miopenFloat, y_backprop, x, scale, mean, variance, in DoBatchNormalizationBackward()
3645 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl() argument
3662 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1214 ShapeHandle y_backprop; in FusedBatchNormGradShape() local
1215 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), rank, &y_backprop)); in FusedBatchNormGradShape()
1223 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape()
1236 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 256 auto y_backprop = in CreateSimpleFusedBatchNormGrad() local
270 y_backprop, x, scale, reserve_space_1, reserve_space_2, in CreateSimpleFusedBatchNormGrad()
699 auto* y_backprop = context.graph_view->GetNode("y_backprop"); in TEST_F() local
700 ASSERT_NE(y_backprop, nullptr); in TEST_F()
702 VerifyRegularFaninMatch(transpose_node1, 0, y_backprop->GetName(), 0); in TEST_F()
|
| D | layout_optimizer_test.cc | 205 Output y_backprop = in SimpleFusedBatchNormGrad() local
221 ops::FusedBatchNormGrad(s->WithOpName("FusedBatchNormGrad"), y_backprop, in SimpleFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1065 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward() argument
1081 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward() argument
|
| D | stream.cc | 371 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x, in ThenBatchNormalizationBackward() argument
379 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc), in ThenBatchNormalizationBackward()
384 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
422 const DeviceMemory<Eigen::half> &y_backprop, in ThenBatchNormalizationBackward() argument
431 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc), in ThenBatchNormalizationBackward()
436 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
|
| D | stream.h | 262 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x,
287 const DeviceMemory<Eigen::half> &y_backprop,
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1943 y_backprop = pfor_input.stacked_input(0)
1944 y_backprop, _, _ = _channel_flatten_input(y_backprop, data_format)
1947 inputs = [y_backprop, x] + [
|