diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-06-03 15:00:34 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-06-04 16:38:40 -0700 |
| commit | cd79f0e06bf53c2c0fee39ee916bb6d79f177b57 (patch) | |
| tree | 367078aeef8fd376711abfe6e52de7bfe491e527 /verif/tosa_serializer.py | |
| parent | 571f7182a10a974f1ce993d83b01070153f142cc (diff) | |
| download | reference_model-cd79f0e06bf53c2c0fee39ee916bb6d79f177b57.tar.gz | |
Rewrite model frontend to be json-driven.
Change-Id: Iac786eff96183938d2fd11cde9313c6e8e1270a5
Diffstat (limited to 'verif/tosa_serializer.py')
| -rw-r--r-- | verif/tosa_serializer.py | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/verif/tosa_serializer.py b/verif/tosa_serializer.py index 5ed9877..3b03252 100644 --- a/verif/tosa_serializer.py +++ b/verif/tosa_serializer.py @@ -645,30 +645,24 @@ class TosaSerializer: test_desc["tosa_file"] = tosa_filename ifm_name = [] - ifm_shape = [] ifm_file = [] ofm_name = [] ofm_file = [] - ofm_shape = [] for b in self.basicBlocks: if b.name == "main": for i in b.inputs: ifm_name.append(i) - ifm_shape.append(b.tensors[i].shape) ifm_file.append(b.tensors[i].placeholderFilename) for o in b.outputs: ofm_name.append(o) - ofm_shape.append(b.tensors[o].shape) # Make up an OFM filename here. One isn't generated until the reference tool is # run, so any name is a good name ofm_file.append("ref-{}.npy".format(o)) - test_desc["ifm_placeholder"] = ifm_name + test_desc["ifm_name"] = ifm_name test_desc["ifm_file"] = ifm_file - test_desc["ifm_shape"] = ifm_shape test_desc["ofm_name"] = ofm_name - test_desc["ofm_shape"] = ofm_shape test_desc["ofm_file"] = ofm_file test_desc["expected_failure"] = self.expectedFailure if self.expectedFailureDesc: |