| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 53 const Tensor& res_sp3, Tensor* output, Tensor* scale_backprop,
162 Tensor scale_backprop; in VerifyTensorsCloseForGrad() local
169 &scale_backprop, &offset_backprop, epsilon); in VerifyTensorsCloseForGrad()
176 ASSERT_EQ(scale_backprop.dtype(), mkl_scale_backprop.dtype()); in VerifyTensorsCloseForGrad()
177 ASSERT_EQ(scale_backprop.shape(), mkl_scale_backprop.shape()); in VerifyTensorsCloseForGrad()
182 test::ExpectClose(scale_backprop, mkl_scale_backprop, 1e-5); in VerifyTensorsCloseForGrad()
385 auto scale_backprop = ops::Identity(root.WithOpName("scale_backprop"), in VerifyFusedBatchNormGradWithConv2D() local
386 bn.scale_backprop); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormGradV3.pbtxt | 32 name: "scale_backprop"
126 name: "scale_backprop"
|
| D | FusedBatchNormGrad.pbtxt | 28 name: "scale_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 28 name: "scale_backprop"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 272 xla::XlaOp scale_backprop; in Compile() local
296 scale_backprop = xla::GetTupleElement(output, 1); in Compile()
333 scale_backprop = xla::Mul(scratch1, scratch2); in Compile()
337 ctx->SetOutput(1, scale_backprop); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 62 Tensor scale_backprop; member
220 backward->scale_backprop = out_tensors[8]; in RunFusedBatchNorm()
332 backward->scale_backprop = out_tensors[8]; in RunFusedBatchNormEx()
407 {fbn_backward.scale_backprop, fbn_ex_backward.scale_backprop}, in VerifyTensorsNear()
|
| D | fused_batch_norm_op.cc | 530 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
606 scale_backprop = scratch2 * scratch1; // DEFAULT DEVICE in operator ()()
988 Tensor* scale_backprop, Tensor* offset_backprop, in operator ()()
1102 StreamExecutorUtil::AsDeviceMemory<U>(*scale_backprop); in operator ()()
1451 Tensor* scale_backprop = nullptr; in ComputeWithReservedSpace() local
1453 &scale_backprop)); in ComputeWithReservedSpace()
1470 f(context->eigen_device<Device>(), scale_backprop->flat<U>()); in ComputeWithReservedSpace()
1478 saved_maybe_inv_var_or_pop_var, epsilon_, x_backprop, scale_backprop, in ComputeWithReservedSpace()
1489 saved_maybe_inv_var_or_pop_var, epsilon_, x_backprop, scale_backprop, in ComputeWithReservedSpace()
|
| D | fused_batch_norm_op.cu.cc | 49 typename TTypes<U>::Vec scale_backprop(scale_backprop_output->vec<U>()); in operator ()() local
104 scale_backprop.device(d) = scratch2 * scratch1; in operator ()()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormGrad.pbtxt | 28 name: "scale_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 28 name: "scale_backprop"
|
| D | FusedBatchNormGradV3.pbtxt | 32 name: "scale_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 | 47 name: "scale_backprop"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 47 name: "scale_backprop"
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 55 name: "scale_backprop"
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 252 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
264 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop,
610 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop,
|
| D | cuda_dnn.cc | 3648 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3655 epsilon, x_backprop, scale_backprop, offset_backprop, in DoBatchNormalizationBackward()
3666 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3673 epsilon, x_backprop, scale_backprop, offset_backprop, in DoBatchNormalizationBackward()
3685 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl() argument
3731 /*dBnScaleData=*/scale_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
3749 scale_backprop->opaque(), offset_backprop->opaque(), epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 290 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop,
302 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop,
725 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop,
|
| D | rocm_dnn.cc | 3616 DeviceMemory<Eigen::half>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3622 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3632 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
3638 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3649 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_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 | 276 auto scale_backprop = ops::Identity(scope.WithOpName("scale_backprop"), in CreateSimpleFusedBatchNormGrad() local
277 fused_batch_norm_grad.scale_backprop); in CreateSimpleFusedBatchNormGrad()
1207 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
1208 ASSERT_NE(scale_backprop, nullptr); in TEST_F()
1209 ASSERT_EQ(scale_backprop->NumRegularFanins(), 1); in TEST_F()
1210 VerifyRegularFaninMatch(scale_backprop, 0, fbng_node->GetName(), 1); in TEST_F()
1270 auto* scale_backprop = context.graph_view->GetNode("scale_backprop"); in TEST_F() local
1271 ASSERT_NE(scale_backprop, nullptr); in TEST_F()
1272 ASSERT_EQ(scale_backprop->NumRegularFanins(), 1); in TEST_F()
1273 VerifyRegularFaninMatch(scale_backprop, 0, fbng_node->GetName(), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1070 DeviceMemory<float>* x_backprop, DeviceMemory<float>* scale_backprop, in DoBatchNormalizationBackward() argument
1087 DeviceMemory<float>* scale_backprop, DeviceMemory<float>* offset_backprop, in DoBatchNormalizationBackward() argument
|
| D | stream.cc | 375 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop, in ThenBatchNormalizationBackward() argument
381 PARAM(scale_backprop), PARAM(offset_backprop)); in ThenBatchNormalizationBackward()
385 epsilon, x_backprop, scale_backprop, offset_backprop, in ThenBatchNormalizationBackward()
427 DeviceMemory<Eigen::half> *x_backprop, DeviceMemory<float> *scale_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 | 266 DeviceMemory<float> *x_backprop, DeviceMemory<float> *scale_backprop,
293 DeviceMemory<float> *scale_backprop, DeviceMemory<float> *offset_backprop,
|