| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 147 xla::XlaOp offset_backprop; in Compile() local
171 offset_backprop = xla::GetTupleElement(output, 2); in Compile()
190 offset_backprop = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
212 ctx->SetOutput(2, offset_backprop); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 172 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
217 offset_backprop.device(d) = y_backprop_sum; in operator ()()
388 Tensor* scale_backprop, Tensor* offset_backprop, in operator ()()
480 StreamExecutorUtil::AsDeviceMemory<U>(*offset_backprop); in operator ()()
653 Tensor* offset_backprop = nullptr; in Compute() local
655 &offset_backprop)); in Compute()
673 f(context->eigen_device<Device>(), offset_backprop->flat<U>()); in Compute()
681 offset_backprop, tensor_format_); in Compute()
699 x_backprop, scale_backprop, offset_backprop, scratch1.vec<U>(), in Compute()
|
| D | fused_batch_norm_op.h | 80 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator()
106 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis); in operator()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 53 name: "offset_backprop"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 53 name: "offset_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 250 DeviceMemory<float>* offset_backprop) override;
260 DeviceMemory<float>* offset_backprop) override;
579 DeviceMemory<U>* offset_backprop);
|
| D | cuda_dnn.cc | 3286 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward() argument
3290 epsilon, x_backprop, scale_backprop, offset_backprop), in DoBatchNormalizationBackward()
3301 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward() argument
3305 epsilon, x_backprop, scale_backprop, offset_backprop), in DoBatchNormalizationBackward()
3317 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl() argument
3338 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 241 DeviceMemory<float>* offset_backprop) override;
251 DeviceMemory<float>* offset_backprop) override;
651 DeviceMemory<U>* offset_backprop);
|
| D | rocm_dnn.cc | 3091 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward() argument
3095 offset_backprop); in DoBatchNormalizationBackward()
3105 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward() argument
3109 offset_backprop); in DoBatchNormalizationBackward()
3120 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl() argument
3135 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1046 DeviceMemory<float>* offset_backprop) {
1061 DeviceMemory<float>* offset_backprop) {
|
| D | stream.cc | 373 DeviceMemory<float> *offset_backprop) { in ThenBatchNormalizationBackward() argument
376 PARAM(scale_backprop), PARAM(offset_backprop)); in ThenBatchNormalizationBackward()
381 epsilon, x_backprop, scale_backprop, offset_backprop)); in ThenBatchNormalizationBackward()
424 DeviceMemory<float> *offset_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 | 251 DeviceMemory<float> *offset_backprop);
274 DeviceMemory<float> *offset_backprop);
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12803 …atchNormGradV2Attr) (x_backprop tf.Output, scale_backprop tf.Output, offset_backprop tf.Output, re…
30505 …dBatchNormGradAttr) (x_backprop tf.Output, scale_backprop tf.Output, offset_backprop tf.Output, re…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 27882 name: "offset_backprop"
27955 name: "offset_backprop"
28034 name: "offset_backprop"
28117 name: "offset_backprop"
28201 name: "offset_backprop"
|
| D | ops_history.v2.pbtxt | 27511 name: "offset_backprop"
27584 name: "offset_backprop"
27663 name: "offset_backprop"
27746 name: "offset_backprop"
27830 name: "offset_backprop"
|
| D | ops_history.v0.pbtxt | 12866 name: "offset_backprop"
12945 name: "offset_backprop"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13822 name: "offset_backprop"
13901 name: "offset_backprop"
|