Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dunfuse_batch_norm.mlir24 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
43 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
58 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
73 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
87 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
129 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
Dlhlo_gpu_ops.mlir17 …"lmhlo_gpu.batch_norm_inference"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg_out) {epsilon = 1.000000e…
Dlhlo_ops.mlir815 …"lmhlo.batch_norm_inference"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg_out) {epsilon = 1.000000e-03 …
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1048 HloInstruction* batch_norm_inference) { in HandleBatchNormInference() argument
1049 return CheckShape(batch_norm_inference, in HandleBatchNormInference()
1051 batch_norm_inference->operand(0)->shape(), in HandleBatchNormInference()
1052 batch_norm_inference->operand(1)->shape(), in HandleBatchNormInference()
1053 batch_norm_inference->operand(2)->shape(), in HandleBatchNormInference()
1054 batch_norm_inference->operand(3)->shape(), in HandleBatchNormInference()
1055 batch_norm_inference->operand(4)->shape(), in HandleBatchNormInference()
1056 batch_norm_inference->feature_index())); in HandleBatchNormInference()
Dhlo_cost_analysis.h99 const HloInstruction* batch_norm_inference) override;
Dhlo_evaluator.h275 HloInstruction* batch_norm_inference) override { in HandleBatchNormInference() argument
Dhlo_verifier.h106 HloInstruction* batch_norm_inference) override;
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.td63 def LHLOGPU_BatchNormInferenceOp : LHLOGPU_Op<"batch_norm_inference">,
Dlhlo_ops.td353 def LHLO_BatchNormInferenceOp : LHLO_Op<"batch_norm_inference", []>,
Dhlo_ops.td767 def HLO_BatchNormInferenceOp : HLO_Op<"batch_norm_inference",
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt437 // CHECK: "lmhlo_gpu.batch_norm_inference"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir16 …// CHECK: "mhlo.batch_norm_inference"(%arg0, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 :…
34 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
53 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
62 …// CHECK: [[Y:%.*]] = "mhlo.batch_norm_inference"([[CONVERT_X]], [[SCALE]], [[OFFSET]], [[MEAN]], …
134 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …