aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/data_nodes.h
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/ops/data_nodes.h')
-rw-r--r--reference_model/src/ops/data_nodes.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/reference_model/src/ops/data_nodes.h b/reference_model/src/ops/data_nodes.h
index a02d441..407cf0a 100644
--- a/reference_model/src/ops/data_nodes.h
+++ b/reference_model/src/ops/data_nodes.h
@@ -24,7 +24,7 @@ namespace TosaReference
class OpConst : public GraphNode
{
public:
- OpConst(uint64_t id_);
+ OpConst(SubgraphTraverser* sgt_, uint64_t id_);
virtual ~OpConst();
virtual int checkTensorAttributes();
@@ -35,7 +35,7 @@ template <int Rank, DType Dtype>
class OpIdentity : public GraphNode
{
public:
- OpIdentity(TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_);
+ OpIdentity(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_);
virtual ~OpIdentity();
virtual int checkTensorAttributes();