Skip to content

Commit

Permalink
Add assert message (apache#11665)
Browse files Browse the repository at this point in the history
Change-Id: I88f19c7105cce048d2f52d50450a551fb12162dc
  • Loading branch information
NicolaLancellotti committed Jun 10, 2022
1 parent f117244 commit e7f793d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/relay/backend/contrib/ethosu/te/identity.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,7 @@ def match_ethosu_identity(output_tensor, device_config):

input_tensors_shape = input_tensors[0].shape
length = len(input_tensors_shape)
assert length <= 4
assert length <= 4, "Input tensor shape must be <= 4 for the identity operator"
channels = int(input_tensors_shape[length - 1]) if length >= 3 else 1

subkernels = len(device_config.get_kernel_steps(identity.op.name, 1, 1, ifm_dtype))
Expand Down

0 comments on commit e7f793d

Please sign in to comment.