colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dbueg

This commit is contained in:
Tian Jin 2019-12-24 03:07:01 -05:00
parent 4eb95b2373
commit 17ecc53bbc
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/mlir/onnx/onnx_lowering.mlir
View File

@ -145,9 +145,9 @@ func @test_exp(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> {
// CHECK: return [[RES]] : memref<?x10xf32> // CHECK: return [[RES]] : memref<?x10xf32>
} }
// func @test_tanh(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> { func @test_tanh(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> {
// %0 = "onnx.Tanh"(%arg0) : (tensor<?x10xf32>) -> tensor<*xf32> %0 = "onnx.Tanh"(%arg0) : (tensor<?x10xf32>) -> tensor<*xf32>
// "std.return"(%0) : (tensor<*xf32>) -> () "std.return"(%0) : (tensor<*xf32>) -> ()
// // CHECK-LABEL: test_tanh // // CHECK-LABEL: test_tanh
// // CHECK: [[DIM_0:%.+]] = dim %arg0, 0 : memref<?x10xf32> // // CHECK: [[DIM_0:%.+]] = dim %arg0, 0 : memref<?x10xf32>
@ -168,7 +168,7 @@ func @test_exp(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> {
// // CHECK: [[TANH_RES:%.+]] = divf [[DIVIDEND]], [[DIVISOR]] : f32 // // CHECK: [[TANH_RES:%.+]] = divf [[DIVIDEND]], [[DIVISOR]] : f32
// // CHECK: store [[TANH_RES]], [[RES]][%arg1, %arg2] : memref<?x10xf32> // // CHECK: store [[TANH_RES]], [[RES]][%arg1, %arg2] : memref<?x10xf32>
// // CHECK: return [[RES]] : memref<?x10xf32> // // CHECK: return [[RES]] : memref<?x10xf32>
// } }
func @test_sinh(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> { func @test_sinh(%arg0 : tensor<?x10xf32>) -> tensor<*xf32> {
%0 = "onnx.Sinh"(%arg0) : (tensor<?x10xf32>) -> tensor<*xf32> %0 = "onnx.Sinh"(%arg0) : (tensor<?x10xf32>) -> tensor<*xf32>