diff options
Diffstat (limited to 'reference_model/src/ops/ewise_binary.h')
| -rw-r--r-- | reference_model/src/ops/ewise_binary.h | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/reference_model/src/ops/ewise_binary.h b/reference_model/src/ops/ewise_binary.h index a5b1059..86b2101 100644 --- a/reference_model/src/ops/ewise_binary.h +++ b/reference_model/src/ops/ewise_binary.h @@ -42,7 +42,7 @@ template <int Rank, DType InDtype, DType OutDtype> class BinaryNodeBase : public GraphNode { public: - BinaryNodeBase(const Op& nodeType, TosaQuantInfoBase* qinfo_, const uint64_t id_); + BinaryNodeBase(SubgraphTraverser* sgt_, const Op& nodeType, TosaQuantInfoBase* qinfo_, const uint64_t id_); virtual ~BinaryNodeBase(); virtual int checkTensorAttributes() final; @@ -76,8 +76,8 @@ template <int Rank, DType InDtype, DType OutDtype> class BinaryNode : public BinaryNodeBase<Rank, InDtype, OutDtype> { public: - BinaryNode(const Op& op_, TosaQuantInfoBase* qinfo_, const uint64_t id_) - : BinaryNodeBase<Rank, InDtype, OutDtype>(op_, qinfo_, id_) + BinaryNode(SubgraphTraverser* sgt_, const Op& op_, TosaQuantInfoBase* qinfo_, const uint64_t id_) + : BinaryNodeBase<Rank, InDtype, OutDtype>(sgt_, op_, qinfo_, id_) {} virtual ~BinaryNode() {} @@ -95,8 +95,8 @@ template <DType InDtype, DType OutDtype> class BinaryNode<0, InDtype, OutDtype> : public BinaryNodeBase<0, InDtype, OutDtype> { public: - BinaryNode(const Op& op_, TosaQuantInfoBase* qinfo_, const uint64_t id_) - : BinaryNodeBase<0, InDtype, OutDtype>(op_, qinfo_, id_) + BinaryNode(SubgraphTraverser* sgt_, const Op& op_, TosaQuantInfoBase* qinfo_, const uint64_t id_) + : BinaryNodeBase<0, InDtype, OutDtype>(sgt_, op_, qinfo_, id_) {} virtual ~BinaryNode() {} @@ -109,8 +109,8 @@ public: class Op##Opname : public BinaryNode<Rank, Dtype, Dtype> \ { \ public: \ - Op##Opname(TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) \ - : BinaryNode<Rank, Dtype, Dtype>(Op_##OPNAME, qinfo_, id_) \ + Op##Opname(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) \ + : BinaryNode<Rank, Dtype, Dtype>(sgt_, Op_##OPNAME, qinfo_, id_) \ { \ register_fcn(); \ } \ @@ -142,8 +142,11 @@ template <int Rank, DType Dtype> class OpArithmeticRightShift : public BinaryNode<Rank, Dtype, Dtype> { public: - OpArithmeticRightShift(TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) - : BinaryNode<Rank, Dtype, Dtype>(Op_ARITHMETIC_RIGHT_SHIFT, qinfo_, id_) + OpArithmeticRightShift(SubgraphTraverser* sgt_, + TosaAttributeBase* attribute_, + TosaQuantInfoBase* qinfo_, + uint64_t id_) + : BinaryNode<Rank, Dtype, Dtype>(sgt_, Op_ARITHMETIC_RIGHT_SHIFT, qinfo_, id_) { INIT_ATTRIBUTE(ArithmeticRightShift); register_fcn(); @@ -160,8 +163,8 @@ template <int Rank, DType InDtype, DType OutDtype> class OpMul : public BinaryNode<Rank, InDtype, OutDtype> { public: - OpMul(TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) - : BinaryNode<Rank, InDtype, OutDtype>(Op_MUL, qinfo_, id_) + OpMul(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_) + : BinaryNode<Rank, InDtype, OutDtype>(sgt_, Op_MUL, qinfo_, id_) { INIT_ATTRIBUTE(Mul); register_fcn(); @@ -180,7 +183,7 @@ template <int Rank, DType InDtype> class OpTable : public GraphNode { public: - OpTable(TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_); + OpTable(SubgraphTraverser* sgt_, TosaAttributeBase* attribute_, TosaQuantInfoBase* qinfo_, uint64_t id_); virtual ~OpTable(); virtual int checkTensorAttributes(); |