Home
last modified time | relevance | path

Searched refs:batch_norm_training (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1038 HloInstruction* batch_norm_training) { in HandleBatchNormTraining() argument
1039 return CheckShape(batch_norm_training, in HandleBatchNormTraining()
1041 batch_norm_training->operand(0)->shape(), in HandleBatchNormTraining()
1042 batch_norm_training->operand(1)->shape(), in HandleBatchNormTraining()
1043 batch_norm_training->operand(2)->shape(), in HandleBatchNormTraining()
1044 batch_norm_training->feature_index())); in HandleBatchNormTraining()
Dhlo_evaluator.h279 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override { in HandleBatchNormTraining() argument
Dhlo_cost_analysis.h97 const HloInstruction* batch_norm_training) override;
Dhlo_verifier.h104 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override;
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.td76 def LHLOGPU_BatchNormTrainingOp : LHLOGPU_Op<"batch_norm_training">,
Dlhlo_ops.td368 def LHLO_BatchNormTrainingOp : LHLO_Op<"batch_norm_training", []>,
Dhlo_ops.td783 def HLO_BatchNormTrainingOp : HLO_Op<"batch_norm_training", [NoSideEffect]>,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir26 …"lmhlo_gpu.batch_norm_training"(%arg0, %arg1, %arg2, %output, %batch_mean, %batch_var) {epsilon = …
Dlhlo_ops.mlir826 …"lmhlo.batch_norm_training"(%arg0, %arg1, %arg2, %output, %batch_mean, %batch_var) {epsilon = 1.00…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt394 // CHECK: "lmhlo_gpu.batch_norm_training"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir93 …%0 = "mhlo.batch_norm_training" (%input, %scale, %offset) {epsilon = 0.001 : f32, feature_index = …
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir41 …// CHECK: %[[RESULT0:.*]] = "mhlo.batch_norm_training"({{.*}}, %arg1, %arg2) {epsilon = 1.000000e-…
73 …// CHECK: %[[RESULT0:.*]] = "mhlo.batch_norm_training"({{.*}}, %arg1, %arg2) {epsilon = 1.000000e-…
85 …// CHECK: %[[RESULT0:.*]] = "mhlo.batch_norm_training"({{.*}}, %arg1, %arg2) {epsilon = 1.000000e-…
94 …// CHECK: %[[RESULT0:.*]] = "mhlo.batch_norm_training"({{.*}}, %arg1, %arg2) {epsilon = 1.000000e-…
127 …// CHECK: "mhlo.batch_norm_training"({{.*}}, %arg1, %arg2) {epsilon = 1.000000e-03 : f32, feature_…