Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
Dunfuse_batch_norm.mlir24 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
77 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
127 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
154 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
180 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
225 %0 = "mhlo.batch_norm_inference"(%x, %scale, %offset, %mean, %variance)
Dops.mlir4094 // mhlo.batch_norm_inference
4098 …%0 = "mhlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001000e-…
4108 …%0 = "mhlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001000e-…
4118 …%0 = "mhlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001000e-…
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1566 HloInstruction* batch_norm_inference) { in HandleBatchNormInference() argument
1567 return CheckShape(batch_norm_inference, in HandleBatchNormInference()
1569 batch_norm_inference->operand(0)->shape(), in HandleBatchNormInference()
1570 batch_norm_inference->operand(1)->shape(), in HandleBatchNormInference()
1571 batch_norm_inference->operand(2)->shape(), in HandleBatchNormInference()
1572 batch_norm_inference->operand(3)->shape(), in HandleBatchNormInference()
1573 batch_norm_inference->operand(4)->shape(), in HandleBatchNormInference()
1574 batch_norm_inference->feature_index())); in HandleBatchNormInference()
Dhlo_cost_analysis.h143 const HloInstruction* batch_norm_inference) override;
Dhlo_evaluator.h351 HloInstruction* batch_norm_inference) override { in HandleBatchNormInference() argument
Dhlo_verifier.h211 HloInstruction* batch_norm_inference) override;
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo/
Dops.mlir866 …"lmhlo.batch_norm_inference"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg_out) {epsilon = 1.000000e-03 …
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
Dops_stablehlo.mlir4069 // stablehlo.batch_norm_inference
4073 …%0 = "stablehlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001…
4083 …%0 = "stablehlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001…
4093 …%0 = "stablehlo.batch_norm_inference" (%input, %scale, %offset, %mean, %variance) {epsilon = 1.001…
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/
Dlhlo_ops.td745 def LHLO_BatchNormInferenceOp : LHLO_Op<"batch_norm_inference", []> {
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-no-tf2xla-fallback.mlir14 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
34 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
45 …// CHECK: [[Y:%.*]] = "mhlo.batch_norm_inference"([[CONVERT_X]], [[SCALE]], [[OFFSET]], [[MEAN]], …
131 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
Dlegalize-tf.mlir21 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
41 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
52 …// CHECK: [[Y:%.*]] = "mhlo.batch_norm_inference"([[CONVERT_X]], [[SCALE]], [[OFFSET]], [[MEAN]], …
138 …// CHECK: "mhlo.batch_norm_inference"({{.*}}, %arg1, %arg2, %arg3, %arg4) {epsilon = 1.000000e-03 …
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.td1245 def StableHLO_BatchNormInferenceOp : StableHLO_Op<"batch_norm_inference",
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td1357 def HLO_BatchNormInferenceOp : HLO_Op<"batch_norm_inference",