| // RUN: kernel-gen-opt %s -split-input-file -verify-diagnostics | |
| func @alloc_raw(%ctx: !tf_framework.op_kernel_context, %size : index) { | |
| // expected-error @+1 {{`dyn_sizes` count 1 does not match dynamic dimensions}} | |
| %buf = tf_framework.alloc_raw(%ctx, %size) : memref<?x10x?xi8> | |
| return | |
| } |