paddle_ops.mlir 366 字节
Newer Older
1 2
// RUN: infrtopt %s | FileCheck %s
// CHECK-LABEL: @ops
Y
Yan Chunwei 已提交
3
func @ops() {
4 5
  %a = pd.feed() {name="input0"} : tensor<?xf32>
  %b = pd.feed() {name="input1"}: tensor<?xf32>
6
  %d = pd.feed() {name="input3"}: !Infrt.lod_tensor<3x4x9xf32, 0>
7
  %c = "pd.matmul"(%a, %b) {transpose_x=true, transpose_y=false} : (tensor<?xf32>, tensor<?xf32>) -> tensor<?xf32>
Y
Yan Chunwei 已提交
8 9 10

  infrt.return
}