diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-10-27 11:57:57 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-11-20 08:52:14 +0000 |
| commit | 8956761a84f413e6f4c9c7d6e4409b145f81c289 (patch) | |
| tree | 3d6918b6ea2eb42a3d45d45098920b371bae8844 /ethosu/vela/graph_optimiser.py | |
| parent | 083f103fe612a88f41495022af89d5a12ea4aded (diff) | |
| download | ethos-u-vela-8956761a84f413e6f4c9c7d6e4409b145f81c289.tar.gz | |
vela: Improve the scaling is equal check
- Improved tensor and scaling query functions
- Fixed bug in convert_batched_fc_to_conv
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Ibc3d14036540f27cf5e993beb2163d3e0f5e5933
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
| -rw-r--r-- | ethosu/vela/graph_optimiser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 35932d49..4f473ddb 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -333,7 +333,7 @@ def convert_batched_fc_shape(op, arch, nng): # There is a preceding Reshape # Compare input of prev_op and input of op, to see if prev_op can be removed ifm_prev_op = prev_op.inputs[0] - if ifm_prev_op.shape == ifm.shape and check_quantized_tens_scaling_equal(ifm_prev_op, ifm.quantization): + if ifm_prev_op.shape == ifm.shape and check_quantized_tens_scaling_equal(ifm_prev_op, ifm): # prev_op can be removed op.set_input_tensor(ifm_prev_op, 0) else: |