![]() |
NeuZephyr
Simple DL Framework
|
This is the complete list of members for nz::graph::ComputeGraph, including all inherited members.
addInput(const Tensor::shape_type &shape, bool requires_grad=false, const std::string &name="default") | nz::graph::ComputeGraph | |
addInput(const Tensor &tensor, const std::string &name="default") | nz::graph::ComputeGraph | |
addInput(InputNode *input, const std::string &name="default") | nz::graph::ComputeGraph | |
addInput(const Tensor::shape_type &shape, Tensor::value_type *data, bool requires_grad, bool host, const std::string &name="default") | nz::graph::ComputeGraph | |
addInput(const Tensor::shape_type &shape, const std::initializer_list< Tensor::value_type > &data, bool requires_grad, const std::string &name="default") | nz::graph::ComputeGraph | |
addNode(NodeType *node, const std::string &name="default") | nz::graph::ComputeGraph | inline |
addNode(const std::string &type, const std::string &input1, const std::string &input2, const std::string &name="default", Args... args) | nz::graph::ComputeGraph | inline |
addOutput(OutputNode *node, const std::string &name="default") | nz::graph::ComputeGraph | |
backward() | nz::graph::ComputeGraph | |
ComputeGraph()=default | nz::graph::ComputeGraph | |
CreateNode(ComputeGraph *graph, const std::string &type, const std::string &name, std::vector< int > pre, const Tensor::shape_type &shape, float *data, bool requires_grad, float *grad) | nz::graph::ComputeGraph | friend |
fill(const std::string &name, Tensor::value_type val) | nz::graph::ComputeGraph | |
fill(const Node *node, Tensor::value_type val) | nz::graph::ComputeGraph | |
fillAll(Tensor::value_type val) const | nz::graph::ComputeGraph | |
forward() | nz::graph::ComputeGraph | |
getLoss() const | nz::graph::ComputeGraph | |
getOutput() const | nz::graph::ComputeGraph | |
getOutputHost() const | nz::graph::ComputeGraph | |
getOutputNode() const | nz::graph::ComputeGraph | |
isSorted() const | nz::graph::ComputeGraph | |
load(const std::string &path) | nz::graph::ComputeGraph | |
nodesList() | nz::graph::ComputeGraph | |
operator<<(std::ostream &os, ComputeGraph &graph) | nz::graph::ComputeGraph | friend |
operator[](const std::string &name) | nz::graph::ComputeGraph | |
print(std::ostream &os) | nz::graph::ComputeGraph | |
randomize(const std::string &name, unsigned long long seed=0) | nz::graph::ComputeGraph | |
randomize(const Node *node, unsigned long long seed=0) | nz::graph::ComputeGraph | |
randomizeAll() const | nz::graph::ComputeGraph | |
save(const std::string &path) | nz::graph::ComputeGraph | |
setInput(const std::string &name, Tensor::value_type *data) | nz::graph::ComputeGraph | |
setInput(const Node *node, Tensor::value_type *data) | nz::graph::ComputeGraph | |
topologicalSort() | nz::graph::ComputeGraph | |
update(Optimizer *optimizer) const | nz::graph::ComputeGraph | |
zeroGrad() const | nz::graph::ComputeGraph | |
~ComputeGraph()=default | nz::graph::ComputeGraph |