| // RUN: tf-mlir-translate -hlo-text-to-mlir-hlo %s -o - | FileCheck %s |
| |
| HloModule main.5 |
| |
| // CHECK-LABEL: func @main(%arg0: tensor<4xf32>, %arg1: tensor<f32>) -> tensor<4xf32> { |
| ENTRY %padding.1 (Arg_0.1: f32[4], Arg_1.2: f32[]) -> f32[4] { |
| %Arg_0.1 = f32[4] parameter(0) |
| %Arg_1.2 = f32[] parameter(1) |
| |
| // CHECK-NEXT: %0 = "xla.pad"(%arg0, %arg1) {edge_padding_high = dense<0> : tensor<1xi64>, edge_padding_low = dense<0> : tensor<1xi64>, interior_padding = dense<0> : tensor<1xi64>} : (tensor<4xf32>, tensor<f32>) -> tensor<4xf32> |
| // CHECK-NEXT: return %0 : tensor<4xf32> |
| ROOT %pad.3 = f32[4] pad(%Arg_0.1, %Arg_1.2), padding=0_0_0 |
| } |
| |
| // CHECK-LABEL: func @padding.2(%arg0: tensor<4x4x4xf32>, %arg1: tensor<f32>) -> tensor<7x11x15xf32> { |
| %padding.2 (Arg_0.1: f32[4, 4, 4], Arg_1.2: f32[]) -> f32[7, 11, 15] { |
| %Arg_0.1 = f32[4, 4, 4] parameter(0) |
| %Arg_1.2 = f32[] parameter(1) |
| |
| // CHECK-NEXT: %0 = "xla.pad"(%arg0, %arg1) {edge_padding_high = dense<[2, 4, 6]> : tensor<3xi64>, edge_padding_low = dense<[1, 3, 5]> : tensor<3xi64>, interior_padding = dense<0> : tensor<3xi64>} : (tensor<4x4x4xf32>, tensor<f32>) -> tensor<7x11x15xf32> |
| // CHECK-NEXT: return %0 : tensor<7x11x15xf32> |
| ROOT %pad.3 = f32[7, 11, 15] pad(%Arg_0.1, %Arg_1.2), padding=1_2x3_4x5_6 |
| } |