Struct ParamNode

• Defined in File node_operators.h

Inheritance Relationships

Base Type

• public marian::Node (Class Node)

Struct Documentation

struct ParamNode : public marian::Node

A parameter node for the graph.

A parameter node is used to store model parameters whose value can be changed during the training, such as weights and biases. To construct a parameter node in the graph, we use param() function in ExpressionGraph class.

Public Functions

ParamNode(Ptr<ExpressionGraph> graph, const Shape &shape, const Ptr<inits::NodeInitializer> &init, bool fixed = false)

ParamNode(Ptr<ExpressionGraph> graph, const Shape &shape, const Ptr<inits::NodeInitializer> &init, Type valueType, bool fixed = false)

~ParamNode()

virtual void allocate()

void init()

const std::string type()

const std::string form()

const std::string color()

virtual size_t hash()

virtual bool equal(Expr node)

virtual void record(Ptr<AutoTunerRecorder>, size_t, bool)