| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 54 Tensor* offset_backprop, bool disable_grappler_opts)>;
163 Tensor offset_backprop; in VerifyTensorsCloseForGrad() local
169 &scale_backprop, &offset_backprop, epsilon); in VerifyTensorsCloseForGrad()
178 ASSERT_EQ(offset_backprop.dtype(), mkl_offset_backprop.dtype()); in VerifyTensorsCloseForGrad()
179 ASSERT_EQ(offset_backprop.shape(), mkl_offset_backprop.shape()); in VerifyTensorsCloseForGrad()
183 test::ExpectClose(offset_backprop, mkl_offset_backprop, 1e-5); in VerifyTensorsCloseForGrad()
387 auto offset_backprop = ops::Identity( in VerifyFusedBatchNormGradWithConv2D() local
388 root.WithOpName("offset_backprop"), bn.offset_backprop); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormGradV3.pbtxt | 36 name: "offset_backprop"
130 name: "offset_backprop"
|
| D | FusedBatchNormGrad.pbtxt | 32 name: "offset_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 32 name: "offset_backprop"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 273 xla::XlaOp offset_backprop; in Compile() local
297 offset_backprop = xla::GetTupleElement(output, 2); in Compile()
316 offset_backprop = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
338 ctx->SetOutput(2, offset_backprop); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 63 Tensor offset_backprop; member
221 backward->offset_backprop = out_tensors[9]; in RunFusedBatchNorm()
333 backward->offset_backprop = out_tensors[9]; in RunFusedBatchNormEx()
408 {fbn_backward.offset_backprop, fbn_ex_backward.offset_backprop}, in VerifyTensorsNear()
|
| D | fused_batch_norm_op.cc | 393 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
478 auto y_backprop_sum = offset_backprop; in operator ()()
988 Tensor* scale_backprop, Tensor* offset_backprop, in operator ()()
1104 StreamExecutorUtil::AsDeviceMemory<U>(*offset_backprop); in operator ()()
1454 Tensor* offset_backprop = nullptr; in ComputeWithReservedSpace() local
1456 &offset_backprop)); in ComputeWithReservedSpace()
1471 f(context->eigen_device<Device>(), offset_backprop->flat<U>()); in ComputeWithReservedSpace()
1479 offset_backprop, use_reserved_space, tensor_format_); in ComputeWithReservedSpace()
1490 offset_backprop); in ComputeWithReservedSpace()
|
| D | fused_batch_norm_op.cu.cc | 50 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local
88 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis); in operator ()()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormGrad.pbtxt | 32 name: "offset_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 32 name: "offset_backprop"
|
| D | FusedBatchNormGradV3.pbtxt | 36 name: "offset_backprop"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | layout_optimization_layout_assignment_gpu_cc_60.mlir | 125 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
151 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
|
| D | layout_optimization_layout_assignment_gpu_cc_70.mlir | 206 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
232 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
|
| D | layout_optimization_layout_assignment_to_nchw.mlir | 213 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
|
| /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"
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 61 name: "offset_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 253 DeviceMemory<float>* offset_backprop,
264 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop,
611 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data,
|
| D | cuda_dnn.cc | 3649 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
3655 epsilon, x_backprop, scale_backprop, offset_backprop, in DoBatchNormalizationBackward()
3667 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
3673 epsilon, x_backprop, scale_backprop, offset_backprop, in DoBatchNormalizationBackward()
3686 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackwardImpl() argument
3732 /*dBnBiasData=*/offset_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
3749 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 291 DeviceMemory<float>* offset_backprop,
302 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop,
726 DeviceMemory<U>* offset_backprop);
|
| D | rocm_dnn.cc | 3617 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
3623 offset_backprop); in DoBatchNormalizationBackward()
3633 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
3639 offset_backprop); in DoBatchNormalizationBackward()
3650 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl() argument
3665 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 278 auto offset_backprop = ops::Identity(scope.WithOpName("offset_backprop"), in CreateSimpleFusedBatchNormGrad() local
279 fused_batch_norm_grad.offset_backprop); in CreateSimpleFusedBatchNormGrad()
1212 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
1213 ASSERT_NE(offset_backprop, nullptr); in TEST_F()
1214 ASSERT_EQ(offset_backprop->NumRegularFanins(), 1); in TEST_F()
1215 VerifyRegularFaninMatch(offset_backprop, 0, fbng_node->GetName(), 2); in TEST_F()
1275 auto* offset_backprop = context.graph_view->GetNode("offset_backprop"); in TEST_F() local
1276 ASSERT_NE(offset_backprop, nullptr); in TEST_F()
1277 ASSERT_EQ(offset_backprop->NumRegularFanins(), 1); in TEST_F()
1278 VerifyRegularFaninMatch(offset_backprop, 0, fbng_node->GetName(), 2); in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1071 DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
1087 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
|
| D | stream.cc | 376 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward() argument
381 PARAM(scale_backprop), PARAM(offset_backprop)); in ThenBatchNormalizationBackward()
385 epsilon, x_backprop, scale_backprop, offset_backprop, in ThenBatchNormalizationBackward()
428 DeviceMemory<float> *offset_backprop, in ThenBatchNormalizationBackward() argument
433 PARAM(scale_backprop), PARAM(offset_backprop)); in ThenBatchNormalizationBackward()
437 epsilon, x_backprop, scale_backprop, offset_backprop, in ThenBatchNormalizationBackward()
|
| D | stream.h | 267 DeviceMemory<float> *offset_backprop,
293 DeviceMemory<float> *scale_backprop, DeviceMemory<float> *offset_backprop,
|