diff options
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
| -rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 20431273..b4a633e9 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -583,6 +583,7 @@ def get_arch_block_config( block_config, arch, block_type, + npu_op.ofm.shape, ifm_shape, ifm2_shape, uses_scalar, |