| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 145 xla::XlaOp x_backprop; in Compile() local
169 x_backprop = xla::GetTupleElement(output, 0); in Compile()
205 x_backprop = in Compile()
210 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
|
| /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
234 x_backprop.reshape(rest_by_depth).device(d) = in operator ()()
387 const Tensor& inv_variance, U epsilon, Tensor* x_backprop, in operator ()()
417 x_backprop_ptr = StreamExecutorUtil::AsDeviceMemory<T>(*x_backprop); in operator ()()
501 x_backprop->tensor<T, 4>()); in operator ()()
645 Tensor* x_backprop = nullptr; in Compute() local
647 context->allocate_output(0, x.shape(), &x_backprop)); in Compute()
680 saved_maybe_inv_var_or_pop_var, epsilon_, x_backprop, scale_backprop, in Compute()
699 x_backprop, scale_backprop, offset_backprop, scratch1.vec<U>(), in Compute()
|
| D | fused_batch_norm_op.h | 78 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator()
118 x_backprop.reshape(rest_by_depth).device(d) = in operator()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 41 name: "x_backprop"
86 The data format for y_backprop, x, x_backprop.
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 41 name: "x_backprop"
92 The data format for y_backprop, x, x_backprop.
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 249 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
258 DeviceMemory<Eigen::half>* x_backprop,
578 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop,
|
| D | cuda_dnn.cc | 3285 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3290 epsilon, x_backprop, scale_backprop, offset_backprop), in DoBatchNormalizationBackward()
3300 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3305 epsilon, x_backprop, scale_backprop, offset_backprop), in DoBatchNormalizationBackward()
3316 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl() argument
3336 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 240 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
249 DeviceMemory<Eigen::half>* x_backprop,
650 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop,
|
| D | rocm_dnn.cc | 3090 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3094 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3104 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3108 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3119 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl() argument
3133 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 217 return output.x_backprop; in SimpleFusedBatchNormGrad()
524 auto x_backprop = SimpleFusedBatchNormGrad(&s, true); in TEST_F() local
525 Output fetch = ops::Identity(s.WithOpName("Fetch"), {x_backprop}); in TEST_F()
538 auto x_backprop = SimpleFusedBatchNormGrad(&s, false); in TEST_F() local
539 Output fetch = ops::Identity(s.WithOpName("Fetch"), {x_backprop}); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1557 x_backprop = outputs[0]
1558 x_backprop = array_ops.reshape(x_backprop, x_reverse_shape)
1559 x_backprop = array_ops.transpose(x_backprop, x_reverse_order)
1562 outputs = [x_backprop] + outputs
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 2478 ShapeHandle x_backprop; in __anon3e672dd83f02() local
2481 c->ReplaceDim(y_backprop, 3, channel_dim, &x_backprop)); in __anon3e672dd83f02()
2484 c->ReplaceDim(y_backprop, 1, channel_dim, &x_backprop)); in __anon3e672dd83f02()
2486 c->set_output(0, x_backprop); in __anon3e672dd83f02()
|
| D | ops.pbtxt | 13814 name: "x_backprop"
13893 name: "x_backprop"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 851 ShapeHandle x_backprop; in FusedBatchNormGradShape() local
853 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
854 c->set_output(0, x_backprop); in FusedBatchNormGradShape()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1045 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
1059 DeviceMemory<Eigen::half>* x_backprop,
|
| D | stream.cc | 372 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward() argument
375 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward()
381 epsilon, x_backprop, scale_backprop, offset_backprop)); in ThenBatchNormalizationBackward()
423 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward() argument
426 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward()
432 epsilon, x_backprop, scale_backprop, offset_backprop)); in ThenBatchNormalizationBackward()
|
| D | stream.h | 250 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop,
272 DeviceMemory<Eigen::half> *x_backprop,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12803 …put, reserve_space_2 tf.Output, optional ...FusedBatchNormGradV2Attr) (x_backprop tf.Output, scale…
30505 …utput, reserve_space_2 tf.Output, optional ...FusedBatchNormGradAttr) (x_backprop tf.Output, scale…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 27503 name: "x_backprop"
27576 name: "x_backprop"
27655 name: "x_backprop"
27738 name: "x_backprop"
27822 name: "x_backprop"
|
| D | ops_history.v1.pbtxt | 27874 name: "x_backprop"
27947 name: "x_backprop"
28026 name: "x_backprop"
28109 name: "x_backprop"
28193 name: "x_backprop"
|
| D | ops_history.v0.pbtxt | 12858 name: "x_backprop"
12937 name: "x_backprop"
|