Go to the documentation of this file.
28 #ifndef NNVM_PASS_FUNCTIONS_H_
29 #define NNVM_PASS_FUNCTIONS_H_
50 ret.
attrs[
"json"] = std::make_shared<any>(json_str);
61 return ret.
GetAttr<std::string>(
"json");
71 return ret.
GetAttr<std::string>(
"graphir");
96 if (shape_inputs.size() != 0) {
97 graph.
attrs[
"shape_inputs"] = std::make_shared<any>(std::move(shape_inputs));
99 if (shape_attr_key.length() != 0) {
100 graph.
attrs[
"shape_attr_key"] = std::make_shared<any>(std::move(shape_attr_key));
102 return ApplyPass(std::move(graph),
"InferShape");
115 if (dtype_inputs.size() != 0) {
116 graph.
attrs[
"dtype_inputs"] = std::make_shared<any>(std::move(dtype_inputs));
118 if (dtype_attr_key.length() != 0) {
119 graph.
attrs[
"dtype_attr_key"] = std::make_shared<any>(std::move(dtype_attr_key));
121 return ApplyPass(std::move(graph),
"InferType");
140 graph.
attrs[
"device_group_attr_key"] = std::make_shared<any>(std::move(device_group_attr_key));
141 graph.
attrs[
"device_assign_map"] = std::make_shared<any>(std::move(device_assign_map));
142 graph.
attrs[
"device_copy_op"] = std::make_shared<any>(std::move(device_copy_op));
143 return ApplyPass(std::move(graph),
"PlaceDevice");
161 Graph graph, std::vector<NodeEntry> ys, std::vector<NodeEntry> xs,
162 std::vector<NodeEntry> ys_out_grad,
163 std::function<
NodeEntry(std::vector<NodeEntry>&& inputs)> aggregate_fun =
nullptr,
164 std::function<
int(
const Node& node)> mirror_fun =
nullptr,
166 std::vector<const Op*> zero_ops = std::vector<const Op*>(),
167 std::string copy_op_str = std::string()) {
168 graph.
attrs[
"grad_ys"] = std::make_shared<any>(std::move(ys));
170 graph.
attrs[
"grad_xs"] = std::make_shared<any>(std::move(xs));
171 graph.
attrs[
"grad_ys_out_grad"] = std::make_shared<any>(std::move(ys_out_grad));
172 if (aggregate_fun !=
nullptr) {
173 graph.
attrs[
"grad_aggregate_fun"] = std::make_shared<any>(aggregate_fun);
176 if (mirror_fun !=
nullptr) {
177 graph.
attrs[
"grad_mirror_fun"] = std::make_shared<any>(mirror_fun);
180 if (attr_hint_fun !=
nullptr) {
181 graph.
attrs[
"attr_hint_fun"] = std::make_shared<any>(attr_hint_fun);
184 if (zero_ops.size()) {
185 graph.
attrs[
"zero_ops"] = std::make_shared<any>(std::move(zero_ops));
188 if (copy_op_str != std::string()) {
189 graph.
attrs[
"copy_op"] = std::make_shared<any>(std::move(copy_op_str));
192 return ApplyPass(std::move(graph),
"Gradient");
197 #endif // NNVM_PASS_FUNCTIONS_H_
Graph PlaceDevice(Graph graph, std::string device_group_attr_key, DeviceAssignMap device_assign_map, std::string device_copy_op)
Place the devices for each operator in the graph.
Definition: pass_functions.h:138
Graph LoadJSON(const std::string &json_str)
Load a graph from JSON string, redirects to "LoadJSON" pass.
Definition: pass_functions.h:48
std::string PrintGraphIR(Graph graph)
Print graph ir.
Definition: pass_functions.h:69
Node represents an operation in a computation graph.
Definition: node.h:143
Symbolic computation graph. This is the intermediate representation for optimization pass.
Definition: graph.h:47
std::vector< TShape > ShapeVector
The result holder of shape of each NodeEntry in the graph.
Definition: graph_attr_types.h:61
Configuration of nnvm as well as basic data structure.
std::unordered_map< std::string, std::shared_ptr< any > > attrs
attributes of a graph Note that attribute is shared pointer and can be shared across graphs.
Definition: graph.h:61
Graph InferType(Graph graph, DTypeVector dtype_inputs, std::string dtype_attr_key="")
Infer types in the graph given the information.
Definition: pass_functions.h:114
Graph InferShape(Graph graph, ShapeVector shape_inputs, std::string shape_attr_key="")
Infer shapes in the graph given the information.
Definition: pass_functions.h:95
Data structures that can appear in graph attributes.
Graph ApplyPass(Graph src, const std::string &pass)
Apply one pass to the graph.
Definition: pass.h:62
Graph Gradient(Graph graph, std::vector< NodeEntry > ys, std::vector< NodeEntry > xs, std::vector< NodeEntry > ys_out_grad, std::function< NodeEntry(std::vector< NodeEntry > &&inputs)> aggregate_fun=nullptr, std::function< int(const Node &node)> mirror_fun=nullptr, std::function< NodeEntry(const NodeEntry &src, const NodeEntry &like)> attr_hint_fun=nullptr, std::vector< const Op * > zero_ops=std::vector< const Op * >(), std::string copy_op_str=std::string())
Get the gradient graph whose outputs are gradients of xs wrt to ys.
Definition: pass_functions.h:160
std::string SaveJSON(Graph graph)
Save a graph to json, redirects to "SaveJSON" pass.
Definition: pass_functions.h:59
Graph OrderMutation(Graph src)
Add control flow dependencies between nodes.
Definition: pass_functions.h:84
const T & GetAttr(const std::string &attr_name) const
Get the immutable attribute from attrs.
Definition: graph.h:230
an entry that represents output data from a node
Definition: node.h:52
std::unordered_map< std::string, int > DeviceAssignMap
The result holder of device of each operator in the graph.
Definition: graph_attr_types.h:112
std::vector< int > DTypeVector
The result holder of type of each NodeEntry in the graph.
Definition: graph_attr_types.h:76
Pass that can be applied to a graph.