diff options
Diffstat (limited to 'ethosu/vela/test/extapi/test_extapi_encode_weights.py')
| -rw-r--r-- | ethosu/vela/test/extapi/test_extapi_encode_weights.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/test/extapi/test_extapi_encode_weights.py b/ethosu/vela/test/extapi/test_extapi_encode_weights.py index 0459cae1..854d14c0 100644 --- a/ethosu/vela/test/extapi/test_extapi_encode_weights.py +++ b/ethosu/vela/test/extapi/test_extapi_encode_weights.py @@ -30,8 +30,8 @@ from ethosu.vela.architecture_features import Accelerator Accelerator.Ethos_U55_64, Accelerator.Ethos_U55_128, Accelerator.Ethos_U55_256, - Accelerator.Yoda_256, - Accelerator.Yoda_512, + Accelerator.Ethos_U65_256, + Accelerator.Ethos_U65_512, ], ) @pytest.mark.parametrize("dilation_x", [1, 2]) |