diff options
Diffstat (limited to 'ethosu/vela/test/testutil.py')
| -rw-r--r-- | ethosu/vela/test/testutil.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/ethosu/vela/test/testutil.py b/ethosu/vela/test/testutil.py index b06008af..82588278 100644 --- a/ethosu/vela/test/testutil.py +++ b/ethosu/vela/test/testutil.py @@ -80,7 +80,9 @@ def create_elemwise_op( return op -def create_op_with_quant_tensors(op_type, ifm_shape, ofm_shape, weights_shape=None, datatype=DataType.uint8): +def create_op_with_quant_tensors( + op_type, ifm_shape, ofm_shape, weights_shape=None, bias_shape=None, datatype=DataType.uint8 +): ifm = Tensor(ifm_shape, datatype, "in") ifm.quantization = default_quant_params() ofm = Tensor(ofm_shape, datatype, "out") @@ -102,6 +104,12 @@ def create_op_with_quant_tensors(op_type, ifm_shape, ofm_shape, weights_shape=No "weights", weights_shape, datatype, np.zeros(weights_shape), np_type, quantization=qp ) op.add_input_tensor(weights) + # Optional bias tensor + if bias_shape is not None: + qp = default_quant_params() + qp.zero_point = np.zeros(bias_shape) + bias = create_const_tensor("bias", bias_shape, DataType.int32, np.zeros(bias_shape), np.int32, quantization=qp) + op.add_input_tensor(bias) return op |