| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 164 typename TTypes<T, 4>::ConstTensor y_backprop( in operator ()() local
213 y_backprop.eval().reshape(rest_by_depth).template cast<U>(); in operator ()()
385 void operator()(OpKernelContext* context, const Tensor& y_backprop, in operator ()()
401 << " y_backprop shape: " << y_backprop.shape().DebugString() in operator ()()
407 Tensor y_backprop_maybe_transformed = y_backprop; in operator ()()
615 const Tensor& y_backprop = context->input(0); in Compute() local
627 OP_REQUIRES(context, y_backprop.dims() == 4, in Compute()
629 y_backprop.shape().DebugString())); in Compute()
679 context, y_backprop, x, scale, saved_mean_or_pop_mean, in Compute()
697 context->eigen_device<Device>(), y_backprop, x, scale, in Compute()
|
| D | fused_batch_norm_op.h | 72 typename TTypes<T, 4>::ConstTensor y_backprop( in operator()
103 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator()
|
| /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.
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 244 Stream* stream, const DeviceMemory<float>& y_backprop,
253 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop,
574 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x,
|
| D | cuda_dnn.cc | 3280 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward() argument
3288 stream, CUDNN_DATA_FLOAT, CUDNN_DATA_FLOAT, y_backprop, in DoBatchNormalizationBackward()
3295 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward() argument
3303 stream, CUDNN_DATA_HALF, CUDNN_DATA_FLOAT, y_backprop, in DoBatchNormalizationBackward()
3312 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl() argument
3335 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 2453 ShapeHandle y_backprop; in __anon3e672dd83f02() local
2454 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &y_backprop)); in __anon3e672dd83f02()
2463 ? c->Dim(y_backprop, 3) in __anon3e672dd83f02()
2464 : c->Dim(y_backprop, 1); in __anon3e672dd83f02()
2481 c->ReplaceDim(y_backprop, 3, channel_dim, &x_backprop)); in __anon3e672dd83f02()
2484 c->ReplaceDim(y_backprop, 1, channel_dim, &x_backprop)); in __anon3e672dd83f02()
|
| D | ops.pbtxt | 13794 name: "y_backprop"
13873 name: "y_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 235 Stream* stream, const DeviceMemory<float>& y_backprop,
244 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop,
646 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x,
|
| D | rocm_dnn.cc | 3085 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop, in DoBatchNormalizationBackward() argument
3093 stream, miopenHalf, miopenFloat, y_backprop, x, scale, mean, inv_var, in DoBatchNormalizationBackward()
3099 Stream* stream, const DeviceMemory<float>& y_backprop, in DoBatchNormalizationBackward() argument
3107 stream, miopenFloat, miopenFloat, y_backprop, x, scale, mean, variance, in DoBatchNormalizationBackward()
3115 const DeviceMemory<T>& y_backprop, const DeviceMemory<T>& x, in DoBatchNormalizationBackwardImpl() argument
3132 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 825 ShapeHandle y_backprop; in FusedBatchNormGradShape() local
826 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &y_backprop)); in FusedBatchNormGradShape()
840 DimensionHandle channel_dim = c->Dim(y_backprop, channel_dim_index); in FusedBatchNormGradShape()
853 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1545 y_backprop = pfor_input.stacked_input(0)
1546 y_backprop, _, _ = _channel_flatten_input(y_backprop, data_format)
1549 inputs = [y_backprop, x] + [
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1040 Stream* stream, const DeviceMemory<float>& y_backprop,
1054 Stream* stream, const DeviceMemory<Eigen::half>& y_backprop,
|
| D | stream.cc | 368 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x, in ThenBatchNormalizationBackward() argument
374 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc), in ThenBatchNormalizationBackward()
380 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
418 const DeviceMemory<Eigen::half> &y_backprop, in ThenBatchNormalizationBackward() argument
425 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc), in ThenBatchNormalizationBackward()
431 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
|
| D | stream.h | 246 const DeviceMemory<float> &y_backprop, const DeviceMemory<float> &x,
267 const DeviceMemory<Eigen::half> &y_backprop,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 199 Output y_backprop = in SimpleFusedBatchNormGrad() local
215 ops::FusedBatchNormGrad(s->WithOpName("FusedBatchNormGrad"), y_backprop, in SimpleFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1353 …argspec: "args=[\'y_backprop\', \'x\', \'scale\', \'reserve_space_1\', \'reserve_space_2\', \'epsi…
1357 …argspec: "args=[\'y_backprop\', \'x\', \'scale\', \'reserve_space_1\', \'reserve_space_2\', \'epsi…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1353 …argspec: "args=[\'y_backprop\', \'x\', \'scale\', \'reserve_space_1\', \'reserve_space_2\', \'epsi…
1357 …argspec: "args=[\'y_backprop\', \'x\', \'scale\', \'reserve_space_1\', \'reserve_space_2\', \'epsi…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12803 func FusedBatchNormGradV2(scope *Scope, y_backprop tf.Output, x tf.Output, scale tf.Output, reserve…
12814 y_backprop, x, scale, reserve_space_1, reserve_space_2,
30505 func FusedBatchNormGrad(scope *Scope, y_backprop tf.Output, x tf.Output, scale tf.Output, reserve_s…
30516 y_backprop, x, scale, reserve_space_1, reserve_space_2,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 27483 name: "y_backprop"
27556 name: "y_backprop"
27635 name: "y_backprop"
27718 name: "y_backprop"
27802 name: "y_backprop"
|
| D | ops_history.v1.pbtxt | 27854 name: "y_backprop"
27927 name: "y_backprop"
28006 name: "y_backprop"
28089 name: "y_backprop"
28173 name: "y_backprop"
|
| D | ops_history.v0.pbtxt | 12838 name: "y_backprop"
12917 name: "y_backprop"
|