aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKevin Cheng <kevin.cheng@arm.com>2021-10-01 14:33:33 -0700
committerKevin Cheng <kevin.cheng@arm.com>2021-10-01 14:33:43 -0700
commit17e9202cfa6511144f651f1b5846a4b10e6f7f5d (patch)
tree160511184c0ece706cf1abed4d361a151be8092d
parentc72b59cc5c1d9251c7794edbeae8fc6b7f30f783 (diff)
downloadreference_model-17e9202cfa6511144f651f1b5846a4b10e6f7f5d.tar.gz
Fix CONST test.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com> Change-Id: Iea91076f3e0673ce5adfcca247d968416023fa58
-rw-r--r--verif/tosa_test_gen.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py
index 7b71cf4..2384e03 100644
--- a/verif/tosa_test_gen.py
+++ b/verif/tosa_test_gen.py
@@ -2041,9 +2041,9 @@ class TosaTestGen:
)
return result_tens
- def build_placeholder(self, op, val):
- # Add an identity op to avoid warning in the reference model
- return self.build_unary(Op.IDENTITY, val)
+ def build_const(self, op, val):
+ self.ser.addOutputTensor(val)
+ return val
# Type Conversion
def build_cast(self, op, val, out_dtype):
@@ -3200,8 +3200,8 @@ class TosaTestGen:
# Data nodes
"const": {
"op": Op.CONST,
- "operands": (1, 0),
- "build_fcn": (build_placeholder, TosaTensorGen.tgBasic, None),
+ "operands": (0, 1),
+ "build_fcn": (build_const, TosaTensorGen.tgBasic, None),
"types": TYPE_FIB,
},
"identity": {