Go to the documentation of this file.
28 #ifndef NNVM_SYMBOLIC_H_
29 #define NNVM_SYMBOLIC_H_
33 #include <unordered_map>
84 void Print(std::ostream& os)
const;
90 Symbol operator[](
size_t index)
const;
118 std::vector<std::string> ListOutputNames()
const;
130 const std::unordered_map<std::string, const Symbol*>& kwargs,
131 const std::string& name);
143 const std::unordered_map<std::string, const Symbol*>& kwargs,
144 const std::string& name)
const;
152 void AddControlDeps(
const Symbol& src);
158 Symbol GetInternals()
const;
164 Symbol GetChildren()
const;
175 void SetAttrs(
const std::vector<std::pair<std::string, std::string> >& attrs);
186 bool GetAttr(
const std::string& key, std::string* out)
const;
196 std::unordered_map<std::string, std::string> ListAttrs(
ListAttrOption option)
const;
204 std::vector<std::tuple<std::string, std::string, std::string> > ListAttrsRecursive()
const;
211 static Symbol CreateFunctor(
const Op* op, std::unordered_map<std::string, std::string> attrs);
223 static Symbol CreateVariable(
const std::string& name);
229 static Symbol CreateGroup(
const std::vector<Symbol>& symbols);
234 #endif // NNVM_SYMBOLIC_H_
void Copy(Tensor< cpu, dim, DType > dst, const Tensor< cpu, dim, DType > &src, Stream< cpu > *stream=NULL)
copy data from one tensor to another, with same shape
Definition: tensor_cpu-inl.h:145
Read only data structure to reference continuous memory region of array. Provide unified view for vec...
Definition: array_view.h:36
ListInputOption
option passed to ListInputNames
Definition: symbolic.h:60
Configuration of nnvm as well as basic data structure.
std::vector< NodeEntry > outputs
output entries contained in the symbol
Definition: symbolic.h:73
Symbol is help class used to represent the operator node in Graph.
Definition: symbolic.h:50
The attributes of the current operation node. Usually are additional parameters like axis,...
Definition: node.h:107
Transpose Compose(const Transpose &lhs, const Transpose &rhs)
ListAttrOption
option passed to ListAttr
Definition: symbolic.h:53
#define NNVM_DLL
NNVM_DLL prefix for windows.
Definition: c_api.h:37
Graph node data structure.
Operator structure.
Definition: op.h:105