28 #ifndef NNVM_SYMBOLIC_H_ 29 #define NNVM_SYMBOLIC_H_ 35 #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> >
205 ListAttrsRecursive()
const;
212 static Symbol CreateFunctor(
const Op* op,
213 std::unordered_map<std::string, std::string> attrs);
225 static Symbol CreateVariable(
const std::string& name);
231 static Symbol CreateGroup(
const std::vector<Symbol>& symbols);
236 #endif // NNVM_SYMBOLIC_H_ Read only data structure to reference continuous memory region of array. Provide unified view for vec...
Definition: array_view.h:36
The attributes of the current operation node. Usually are additional parameters like axis...
Definition: node.h:119
std::vector< NodeEntry > outputs
output entries contained in the symbol
Definition: symbolic.h:73
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:146
Graph node data structure.
ListInputOption
option passed to ListInputNames
Definition: symbolic.h:60
#define NNVM_DLL
NNVM_DLL prefix for windows.
Definition: c_api.h:37
ListAttrOption
option passed to ListAttr
Definition: symbolic.h:53
Symbol is help class used to represent the operator node in Graph.
Definition: symbolic.h:50
Configuration of nnvm as well as basic data structure.
Operator structure.
Definition: op.h:103