2019-11-19 08:37:58 +08:00
|
|
|
// RUN: onnf-opt --canonicalize %s -split-input-file | FileCheck %s
|
|
|
|
|
|
|
|
//CHECK: module {
|
|
|
|
module {
|
|
|
|
func @test_sigmoid() {
|
|
|
|
%0 = "frontend.input t1"() : () -> tensor<10x10xf32>
|
|
|
|
%1 = "frontend.input t2"() : () -> tensor<10x10xf32>
|
|
|
|
%2 = "frontend.input t3"() : () -> tensor<10x10xf32>
|
|
|
|
// CHECK: %{{[0-9]+}} = "onnx.full_gemm"(%{{.*}}, %{{.*}}, %{{.*}}) : (tensor<10x10xf32>, tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
|
2019-11-19 10:08:21 +08:00
|
|
|
%3 = "onnx.MatMul"(%0, %1) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
|
|
|
|
%4 = "onnx.Add"(%3, %2) : (tensor<10x10xf32>, tensor<10x10xf32>) -> tensor<10x10xf32>
|
2019-11-19 08:37:58 +08:00
|
|
|
%5 = "frontend.output t4"(%4) : (tensor<10x10xf32>) -> tensor<10x10xf32>
|
|
|
|
}
|
2019-11-19 10:08:21 +08:00
|
|
|
}
|