| // RUN: mlir-opt --test-constant-fold %s | FileCheck %s |
| |
| // CHECK-LABEL: func @test_const |
| func.func @test_const(%arg0 : index) -> tensor<4xi32> { |
| // CHECK: tosa.const |
| %0 = "tosa.const"() {value = dense<[3, 0, 1, 2]> : tensor<4xi32>} : () -> tensor<4xi32> |
| return %0 : tensor<4xi32> |
| } |
| |
| // CHECK-LABEL: func @test_const_i64 |
| func.func @test_const_i64(%arg0 : index) -> tensor<4xi64> { |
| // CHECK: tosa.const |
| %0 = "tosa.const"() {value = dense<[3, 0, 1, 2]> : tensor<4xi64>} : () -> tensor<4xi64> |
| return %0 : tensor<4xi64> |
| } |
| |
| // CHECK-LABEL: func @try_fold_equal_with_unranked_tensor |
| func.func @try_fold_equal_with_unranked_tensor(%arg0: tensor<4xi32>, %arg1: tensor<i32>) { |
| // CHECK: tosa.equal |
| // CHECK-NEXT: return |
| %0 = tosa.equal %arg0, %arg1 : (tensor<4xi32>, tensor<i32>) -> tensor<*xi1> |
| return |
| } |