diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-11 17:19:54 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-22 15:57:43 +0100 |
commit | ae2d553c4f3dd71a1df6c0e8c9cb920ae584b59e (patch) | |
tree | 0b90f57e65cbc88b52b51b57cb9cc8230fa41ad2 /ethosu/vela/test/testutil.py | |
parent | 016b827ad722aecd4338d1d6c7b1b004760490b7 (diff) | |
download | ethos-u-vela-ae2d553c4f3dd71a1df6c0e8c9cb920ae584b59e.tar.gz |
MLBEDSW-3499: Support for PAD operator
Replaces the PAD operator by hardware padding when possible.
Change-Id: I9dce0885e51a4a73715824d7368637222e39b2b3
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/test/testutil.py')
-rw-r--r-- | ethosu/vela/test/testutil.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/test/testutil.py b/ethosu/vela/test/testutil.py index c3459501..96aeb7eb 100644 --- a/ethosu/vela/test/testutil.py +++ b/ethosu/vela/test/testutil.py @@ -115,8 +115,9 @@ def create_op_with_quant_tensors( def create_op(op_type, inputs, output, attrs=None): op = Operation(op_type, output.name + "_op") - op.inputs = inputs - op.outputs = [output] + for input in inputs: + op.add_input_tensor(input) + op.set_output_tensor(output) if attrs is not None: op.attrs = attrs op.set_ifm_ofm_shapes() |