| /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
138 func.return %x_backprop : tensor<1x28x28x64xf32>
151 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
164 func.return %x_backprop : tensor<1x28x28x64xf16>
|
| D | layout_optimization_layout_assignment_gpu_cc_70.mlir | 206 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
219 func.return %x_backprop : tensor<1x28x28x64xf32>
232 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
245 func.return %x_backprop : tensor<1x64x28x28xf16>
|
| D | layout_optimization_layout_assignment_to_nchw.mlir | 210 // CHECK-SAME: (%x_backprop, %[[RES_PERM]])
213 %x_backprop, %scale_backprop, %offset_backprop, %reserve_1, %reserve_2
226 func.return %x_backprop : tensor<1x28x28x64xf32>
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 275 xla::XlaOp x_backprop; in Compile() local
299 x_backprop = xla::GetTupleElement(output, 0); in Compile()
335 x_backprop = in Compile()
340 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormGradV3.pbtxt | 28 name: "x_backprop"
122 name: "x_backprop"
|
| D | FusedBatchNormGrad.pbtxt | 24 name: "x_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 24 name: "x_backprop"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 41 name: "x_backprop"
86 The data format for y_backprop, x, x_backprop.
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 41 name: "x_backprop"
92 The data format for y_backprop, x, x_backprop.
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 49 name: "x_backprop"
100 The data format for y_backprop, x, x_backprop.
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 412 typename TTypes<T, 4>::Tensor x_backprop( in operator ()() local
516 x_backprop.reshape(rest_by_depth).device(d) = in operator ()()
544 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator ()() local
611 x_backprop.reshape(rest_by_depth).device(d) = in operator ()()
1000 Tensor* x_backprop, Tensor* scale_backprop, in operator ()()
1050 x_backprop_ptr = StreamExecutorUtil::AsDeviceMemory<T>(*x_backprop); in operator ()()
1170 x_backprop->tensor<T, 4>()); in operator ()()
1574 Tensor* x_backprop = nullptr; in ComputeWithReservedSpace() local
1577 context->allocate_output(0, alloc_shape, &x_backprop)); in ComputeWithReservedSpace()
1614 x_backprop, scale_backprop, offset_backprop, side_input_backprop, in ComputeWithReservedSpace()
[all …]
|
| D | fused_batch_norm_ex_op_test.cc | 61 Tensor x_backprop; member
218 backward->x_backprop = out_tensors[6]; in RunFusedBatchNorm()
352 backward->x_backprop = out_tensors[6]; in RunFusedBatchNormEx()
427 {fbn_backward.x_backprop, fbn_ex_backward.x_backprop}, in VerifyTensorsNear()
|
| D | fused_batch_norm_op.cu.cc | 49 typename TTypes<T, 4>::Tensor x_backprop(x_backprop_output->tensor<T, 4>()); in operator ()() local
107 x_backprop.reshape(rest_by_depth).device(d) = in operator ()()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormGrad.pbtxt | 24 name: "x_backprop"
|
| D | FusedBatchNormGradV3.pbtxt | 28 name: "x_backprop"
|
| D | FusedBatchNormGradV2.pbtxt | 24 name: "x_backprop"
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 315 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop,
329 DeviceMemory<Eigen::half>* x_backprop,
560 dnn::ActivationMode activation_mode, DeviceMemory<T>* x_backprop,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 386 auto x_backprop = in VerifyFusedBatchNormGradWithConv2D() local
387 ops::Identity(root.WithOpName("x_backprop"), bn.x_backprop); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 319 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop,
333 DeviceMemory<Eigen::half>* x_backprop,
662 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop,
|
| D | rocm_dnn.cc | 3733 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward() argument
3740 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3751 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward() argument
3758 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3769 DeviceMemory<T>* x_backprop, DeviceMemory<U>* scale_backprop, in DoBatchNormalizationBackwardImpl() argument
3783 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 274 auto x_backprop = ops::Identity(scope.WithOpName("x_backprop"), in CreateSimpleFusedBatchNormGrad() local
275 fused_batch_norm_grad.x_backprop); in CreateSimpleFusedBatchNormGrad()
1202 auto* x_backprop = context.graph_view->GetNode("x_backprop"); in TEST_F() local
1203 ASSERT_NE(x_backprop, nullptr); in TEST_F()
1204 ASSERT_EQ(x_backprop->NumRegularFanins(), 1); in TEST_F()
1205 VerifyRegularFaninMatch(x_backprop, 0, output_transpose_node->GetName(), 0); in TEST_F()
1265 auto* x_backprop = context.graph_view->GetNode("x_backprop"); in TEST_F() local
1266 ASSERT_NE(x_backprop, nullptr); in TEST_F()
1267 ASSERT_EQ(x_backprop->NumRegularFanins(), 1); in TEST_F()
1268 VerifyRegularFaninMatch(x_backprop, 0, fbng_node->GetName(), 0); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 361 dnn::ActivationMode activation_mode, DeviceMemory<float> *x_backprop, in ThenBatchNormalizationBackward() argument
367 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward()
372 scale_offset_desc, epsilon, activation_mode, x_backprop, scale_backprop, in ThenBatchNormalizationBackward()
416 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *x_backprop, in ThenBatchNormalizationBackward() argument
422 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward()
427 scale_offset_desc, epsilon, activation_mode, x_backprop, scale_backprop, in ThenBatchNormalizationBackward()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-no-tf2xla-fallback.mlir | 185 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[mul3]] : tensor<8x8x8x8xf32>
186 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
199 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[grad_operand]] : tensor<8x8x8x8xf32>
200 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
238 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[mul3]] : tensor<8x8x8x8xf32>
239 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
252 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[grad_operand]] : tensor<8x8x8x8xf32>
253 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
266 …// CHECK: %[[x_backprop:.*]] = mhlo.convert({{.*}}) : (tensor<8x8x8x8xf32>) -> tensor<8x8x8x8xbf16>
267 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xbf16>
[all …]
|
| D | legalize-tf.mlir | 192 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[mul3]] : tensor<8x8x8x8xf32>
193 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
206 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[grad_operand]] : tensor<8x8x8x8xf32>
207 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
245 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[mul3]] : tensor<8x8x8x8xf32>
246 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
259 // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[grad_operand]] : tensor<8x8x8x8xf32>
260 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
273 …// CHECK: %[[x_backprop:.*]] = mhlo.convert({{.*}}) : (tensor<8x8x8x8xf32>) -> tensor<8x8x8x8xbf16>
274 // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xbf16>
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1973 x_backprop = outputs[0]
1974 x_backprop = array_ops.reshape(x_backprop, x_reverse_shape)
1975 x_backprop = array_ops.transpose(x_backprop, x_reverse_order)
1978 outputs = [x_backprop] + outputs
|