diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-10-27 12:43:14 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-11-20 12:48:57 +0000 |
| commit | c8a73868d40cf63380f634baeb51aa7aa993fc0c (patch) | |
| tree | 09ea718df7470068d060fb97d46dafd0878201c4 /ethosu/vela/weight_compressor.py | |
| parent | c7187434c11151a6a03f252c8718f3bf6445ef5b (diff) | |
| download | ethos-u-vela-c8a73868d40cf63380f634baeb51aa7aa993fc0c.tar.gz | |
vela: Rename Yoda to Ethos-U65
- Also changed to use Ethos-U where appropriate
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Ie45ba2bb3935b305abe897b78b498681296cb7c1
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
| -rw-r--r-- | ethosu/vela/weight_compressor.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index c07229fb..0eab1851 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -57,13 +57,13 @@ def encode_weights( block_traversal: NpuBlockTraversal, ): """ - Public facing API to use the ethosu weight encoding. + Public facing API to use the Ethos-U weight encoding. - :param accelerator: architecture_features.Accelerator enum to pick the correct ethosu accelerator + :param accelerator: architecture_features.Accelerator enum to pick the correct Ethos-U accelerator :param weights_volume: numpy.ndarray in OHWI layout with a shape of four :param dilation_xy: a two element tuple of dilation attributes in x,y dimension :param ifm_bitdepth: the bitdepth of input feature map - :param ofm_block_depth: the depth of blocks for ethosu processing + :param ofm_block_depth: the depth of blocks for Ethos-U processing :param is_depthwise: a boolean indicating these weights are used for a depthwise traversal :param block_traversal: indicates how these weights are traversed on sub-kernal basis :return: a bytearray of compressed weights @@ -108,7 +108,8 @@ def encode_weights( def encode_bias(bias: np.int64, scale: int, shift: int): """ - Public facing API to pack bias and scale values as required by the hardware + Public facing API to pack bias and scale values as required by the Ethos-U + :param bias: 64bit signed number that includes 40bit signed bias :param scale: 32bit scale value :param shift: 6bit shift value |