From: Vladimir Plazun/AI Tools Lab/Engineer/삼성전자 Date: Fri, 1 Jun 2018 12:25:23 +0000 (+0400) Subject: Add BiasAdd operation description class (#271) X-Git-Tag: nncc_backup~2639 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=af4811afbd19b3eac7cdbbca924f6046003f5b8e;p=platform%2Fcore%2Fml%2Fnnfw.git Add BiasAdd operation description class (#271) Add BiasAdd operation description class This class used to represent bias addition in computation graph Signed-off-by: Vladimir Plazun --- diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/bias_add_op.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/bias_add_op.h new file mode 100644 index 0000000..ba1255a --- /dev/null +++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/bias_add_op.h @@ -0,0 +1,38 @@ +#ifndef _NNC_CORE_IR_MODEL_BIAS_ADD_H_ +#define _NNC_CORE_IR_MODEL_BIAS_ADD_H_ + +#include "nnc/core/IR/model/operations/operation.h" +#include "nnc/core/linalg/TensorVariant.h" + +namespace nncc +{ +namespace contrib +{ +namespace core +{ +namespace IR +{ +namespace model +{ +namespace ops +{ + +class BiasAddOp : public OpDescription +{ +public: + explicit BiasAddOp(const ADT::TensorVariant &weights) : OpDescription(1, 1), _weights(weights) {} + + const ADT::TensorVariant &getWeights() const { return _weights; } + +private: + ADT::TensorVariant _weights; +}; + +} // namespace ops +} // namespace model +} // namespace IR +} // namespace core +} // namespace contrib +} // namespace nncc + +#endif //_NNC_CORE_IR_MODEL_BIAS_ADD_H_