| // RUN: tf-mlir-translate -hlo-text-to-mlir-hlo %s -o - | FileCheck %s |
| |
| HloModule foo.5 |
| |
| // CHECK-LABEL: func @main(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>) -> tensor<f32> { |
| ENTRY %main.5 (Arg_0.1: f32[4], Arg_1.2: f32[4]) -> f32[] { |
| %Arg_0.1 = f32[4]{0} parameter(0) |
| %Arg_1.2 = f32[4]{0} parameter(1) |
| |
| // CHECK-NEXT: %0 = "xla.add"(%arg0, %arg1) {name = "add.3"} : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> |
| %add.3 = f32[4]{0} add(f32[4]{0} %Arg_0.1, f32[4]{0} %Arg_1.2) |
| |
| // TODO(b/129709049) consider making this default precision config inferred. |
| // CHECK-NEXT: %1 = "xla.dot"(%0, %arg1) {name = "dot.4", precision_config = ["DEFAULT", "DEFAULT"]} : (tensor<4xf32>, tensor<4xf32>) -> tensor<f32> |
| // CHECK-NEXT: return %1 : tensor<f32> |
| ROOT %dot.4 = f32[] dot(f32[4]{0} %add.3, f32[4]{0} %Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={0} |
| } |