tensor_shape.mlir 204 字节
Newer Older
1 2
// RUN: infrtexec -i %s | FileCheck %s
// CHECK-LABEL: @build_tensor1
Y
Yan Chunwei 已提交
3 4
func @build_tensor1() {
  %a = ts.build_shape [1:i64, 57:i64, 92:i64]
5
  // CHECK: shape[1,57,92]
Y
Yan Chunwei 已提交
6 7 8
  ts.print_shape %a
  infrt.return
}