diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-02 18:04:27 +0100 |
|---|---|---|
| committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-23 13:39:35 +0100 |
| commit | aeae56770f3c19182d32cc63fd32396e061a7648 (patch) | |
| tree | 95ca2e6c90d81ba8910c8ca9ced68ffa132b7dab /ethosu/vela/test/extapi/test_extapi_encode_bias.py | |
| parent | 27d36f003d35413beb51c1de8f33259ddeca7543 (diff) | |
| download | ethos-u-vela-aeae56770f3c19182d32cc63fd32396e061a7648.tar.gz | |
MLBEDSW-3424: Expose API through separate file
All external APIs are now exposed by api.py.
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Change-Id: I33f480e424692ac30e9c7d791f583199f31164a7
Diffstat (limited to 'ethosu/vela/test/extapi/test_extapi_encode_bias.py')
| -rw-r--r-- | ethosu/vela/test/extapi/test_extapi_encode_bias.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/test/extapi/test_extapi_encode_bias.py b/ethosu/vela/test/extapi/test_extapi_encode_bias.py index ffdd3b0c..c0a4a9ab 100644 --- a/ethosu/vela/test/extapi/test_extapi_encode_bias.py +++ b/ethosu/vela/test/extapi/test_extapi_encode_bias.py @@ -14,12 +14,12 @@ # See the License for the specific language governing permissions and # limitations under the License. # Description: -# Contains unit tests for encode_biases API for an external consumer +# Contains unit tests for npu_encode_bias API for an external consumer import random import numpy as np -from ethosu.vela.weight_compressor import encode_bias +from ethosu.vela.api import npu_encode_bias def test_encode_bias(): @@ -34,6 +34,6 @@ def test_encode_bias(): bias = np.int64(random.randint(bias_lower_limit, bias_upper_limit)) scale = int(random.randint(scale_lower_limit, scale_upper_limit)) shift = int(random.randint(shift_lower_limit, shift_upper_limit)) - biases_enc = encode_bias(bias, scale, shift) + biases_enc = npu_encode_bias(bias, scale, shift) assert isinstance(biases_enc, bytearray) assert len(biases_enc) == 10 |