| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 146 xla::XlaOp scale_backprop; in Compile() local
170 scale_backprop = xla::GetTupleElement(output, 1); in Compile()
207 scale_backprop = xla::Mul(scratch1, scratch2); in Compile()
211 ctx->SetOutput(1, scale_backprop); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 171 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
214 scale_backprop.device(d) = in operator ()()
388 Tensor* scale_backprop, Tensor* offset_backprop, in operator ()()
478 StreamExecutorUtil::AsDeviceMemory<U>(*scale_backprop); in operator ()()
650 Tensor* scale_backprop = nullptr; in Compute() local
652 &scale_backprop)); in Compute()
672 f(context->eigen_device<Device>(), scale_backprop->flat<U>()); 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 | 79 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator()
124 scale_backprop.device(d) = scratch2 * scratch1; in operator()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 47 name: "scale_backprop"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 47 name: "scale_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 249 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
259 DeviceMemory<float>* scale_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
3338 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 240 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
250 DeviceMemory<float>* scale_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
3135 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1045 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
1060 DeviceMemory<float>* scale_backprop,
|
| D | stream.cc | 372 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward() argument
376 PARAM(scale_backprop), PARAM(offset_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
427 PARAM(scale_backprop), PARAM(offset_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,
273 DeviceMemory<float> *scale_backprop,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12803 …Output, optional ...FusedBatchNormGradV2Attr) (x_backprop tf.Output, scale_backprop tf.Output, off…
30505 …f.Output, optional ...FusedBatchNormGradAttr) (x_backprop tf.Output, scale_backprop tf.Output, off…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 27507 name: "scale_backprop"
27580 name: "scale_backprop"
27659 name: "scale_backprop"
27742 name: "scale_backprop"
27826 name: "scale_backprop"
|
| D | ops_history.v1.pbtxt | 27878 name: "scale_backprop"
27951 name: "scale_backprop"
28030 name: "scale_backprop"
28113 name: "scale_backprop"
28197 name: "scale_backprop"
|
| D | ops_history.v0.pbtxt | 12862 name: "scale_backprop"
12941 name: "scale_backprop"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13818 name: "scale_backprop"
13897 name: "scale_backprop"
|