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