Home
last modified time | relevance | path

Searched refs:batch_norm_grad (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1059 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad() argument
1060 return CheckShape(batch_norm_grad, ShapeInference::InferBatchNormGradShape( in HandleBatchNormGrad()
1061 batch_norm_grad->operand(0)->shape(), in HandleBatchNormGrad()
1062 batch_norm_grad->operand(1)->shape(), in HandleBatchNormGrad()
1063 batch_norm_grad->operand(2)->shape(), in HandleBatchNormGrad()
1064 batch_norm_grad->operand(3)->shape(), in HandleBatchNormGrad()
1065 batch_norm_grad->operand(4)->shape(), in HandleBatchNormGrad()
1066 batch_norm_grad->feature_index())); in HandleBatchNormGrad()
Dhlo_evaluator.h271 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad() argument
Dhlo_cost_analysis.h100 Status HandleBatchNormGrad(const HloInstruction* batch_norm_grad) override;
Dhlo_verifier.h107 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override;
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.td47 def LHLOGPU_BatchNormGradOp : LHLOGPU_Op<"batch_norm_grad">,
Dlhlo_ops.td335 def LHLO_BatchNormGradOp : LHLO_Op<"batch_norm_grad", []>,
Dhlo_ops.td751 def HLO_BatchNormGradOp : HLO_Op<"batch_norm_grad", [NoSideEffect]>,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir8 …"lmhlo_gpu.batch_norm_grad"(%arg0, %arg1, %arg2, %arg3, %arg4, %grad_operand, %grad_scale, %grad_o…
Dlhlo_ops.mlir804 …"lmhlo.batch_norm_grad"(%arg0, %arg1, %arg2, %arg3, %arg4, %grad_operand, %grad_scale, %grad_offse…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt415 // CHECK: "lmhlo_gpu.batch_norm_grad"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir201 …// CHECK-NEXT: %[[training:.*]] = "mhlo.batch_norm_grad"(%[[act]], %arg2, %arg3, %arg4, %[[grad]])…
261 …// CHECK-NEXT: %[[training:.*]] = "mhlo.batch_norm_grad"(%[[act]], %arg2, %arg3, %arg4, %[[grad]])…
288 …// CHECK-NEXT: %[[training:.*]] = "mhlo.batch_norm_grad"(%[[act]], %arg2, %arg3, %arg4, %[[grad]])…
348 …// CHECK-NEXT: %[[training:.*]] = "mhlo.batch_norm_grad"(%[[act]], %arg2, %arg3, %arg4, %[[grad]])…
376 …// CHECK-NEXT: %[[training:.*]] = "mhlo.batch_norm_grad"(%[[act]], %arg2, %arg3, %arg4, %[[grad]])…
434 …// CHECK: %{{.*}} = "mhlo.batch_norm_grad"(%{{.*}}, %arg2, %arg3, %arg4, %[[grad]]) {epsilon = 1.0…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir76 …%0 = "mhlo.batch_norm_grad" (%input, %scale, %mean, %variance, %grad_output) {epsilon = 0.001 : f3…
Dimport.hlotxt95 // CHECK: "mhlo.batch_norm_grad"