diff options
author | Dwight Lidman <dwight.lidman@arm.com> | 2021-03-15 19:06:10 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-03-16 16:29:01 +0000 |
commit | 9b3791817c32529dfeddae57c29c2abe19311fc4 (patch) | |
tree | 866c769bb8627b686f7d25ddce3413380c34cbdb /ethosu/vela/tensor.py | |
parent | c822d62ba27b874a130e9d8d434c12b419d10d62 (diff) | |
download | ethos-u-vela-9b3791817c32529dfeddae57c29c2abe19311fc4.tar.gz |
MLBEDSW-4215: Add support for MEAN to match QuantizedMeanOrSum implementation
This commit adds support for emulating the behavior
of the QuantizedMeanOrSum implementation of MEAN in
TensorFlow Lite.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Ifd24e0e678e2f85cd66ab82deeaaf010d5351b1e
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 97885d0e..e915363c 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -300,13 +300,16 @@ def create_const_tensor( value_dtype: np.dtype = None, purpose: TensorPurpose = TensorPurpose.Unknown, quantization: QuantizationParameters = None, + quant_value_dtype: np.dtype = None, ): # Tensor const_tensor = Tensor(shape, dtype, name + "_0") const_tensor.purpose = purpose const_tensor.quantization = quantization const_tensor.values = np.array(values, dtype=value_dtype) - const_tensor.quant_values = np.frombuffer(const_tensor.values.tobytes(), dtype=np.uint8) + const_tensor.quant_values = np.frombuffer( + const_tensor.values.tobytes(), dtype=np.uint8 if not quant_value_dtype else quant_value_dtype + ) # Operator const_op = Operation(Op.Const, name) const_op.set_output_tensor(const_tensor) |