29 #ifndef NNVM_SYMBOLIC_H_ 30 #define NNVM_SYMBOLIC_H_ 36 #include <unordered_map> 85 void Print(std::ostream &os)
const;
91 Symbol operator[] (
size_t index)
const;
119 std::vector<std::string> ListOutputNames()
const;
131 const std::unordered_map<std::string, const Symbol*>& kwargs,
132 const std::string& name);
144 const std::unordered_map<std::string, const Symbol*>& kwargs,
145 const std::string& name)
const;
153 void AddControlDeps(
const Symbol& src);
159 Symbol GetInternals()
const;
165 Symbol GetChildren()
const;
176 void SetAttrs(
const std::vector<std::pair<std::string, std::string> >& attrs);
187 bool GetAttr(
const std::string& key, std::string* out)
const;
197 std::unordered_map<std::string, std::string> ListAttrs(
ListAttrOption option)
const;
205 std::vector<std::tuple<std::string, std::string, std::string> >
206 ListAttrsRecursive()
const;
213 static Symbol CreateFunctor(
const Op* op,
214 std::unordered_map<std::string, std::string> attrs);
226 static Symbol CreateVariable(
const std::string& name);
232 static Symbol CreateGroup(
const std::vector<Symbol>& symbols);
237 #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
Graph node data structure.
The attributes of the current operation node. Usually are additional parameters like axis...
Definition: node.h:120
std::vector< NodeEntry > outputs
output entries contained in the symbol
Definition: symbolic.h:74
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:127
ListInputOption
option passed to ListInputNames
Definition: symbolic.h:61
#define NNVM_DLL
NNVM_DLL prefix for windows.
Definition: c_api.h:38
ListAttrOption
option passed to ListAttr
Definition: symbolic.h:54
Symbol is help class used to represent the operator node in Graph.
Definition: symbolic.h:51
Configuration of nnvm as well as basic data structure.
Operator structure.
Definition: op.h:104