NeuZephyr
Simple DL Framework
nz::graph::ComputeGraph Member List

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::ComputeGraphinline
addNode(const std::string &type, const std::string &input1, const std::string &input2, const std::string &name="default", Args... args)nz::graph::ComputeGraphinline
addOutput(OutputNode *node, const std::string &name="default")nz::graph::ComputeGraph
backward()nz::graph::ComputeGraph
ComputeGraph()=defaultnz::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::ComputeGraphfriend
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) constnz::graph::ComputeGraph
forward()nz::graph::ComputeGraph
getLoss() constnz::graph::ComputeGraph
getOutput() constnz::graph::ComputeGraph
getOutputHost() constnz::graph::ComputeGraph
getOutputNode() constnz::graph::ComputeGraph
isSorted() constnz::graph::ComputeGraph
load(const std::string &path)nz::graph::ComputeGraph
nodesList()nz::graph::ComputeGraph
operator<<(std::ostream &os, ComputeGraph &graph)nz::graph::ComputeGraphfriend
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() constnz::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) constnz::graph::ComputeGraph
zeroGrad() constnz::graph::ComputeGraph
~ComputeGraph()=defaultnz::graph::ComputeGraph