Pass functions that simply redirect the calls to ApplyPass. More...
#include <string>#include <memory>#include <vector>#include <utility>#include "base.h"#include "pass.h"#include "graph_attr_types.h"
Include dependency graph for pass_functions.h:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Namespaces | |
| nnvm | |
| nnvm::pass | |
Functions | |
| Graph | nnvm::pass::LoadJSON (const std::string &json_str) |
| Load a graph from JSON string, redirects to "LoadJSON" pass. More... | |
| std::string | nnvm::pass::SaveJSON (Graph graph) |
| Save a graph to json, redirects to "SaveJSON" pass. More... | |
| std::string | nnvm::pass::PrintGraphIR (Graph graph) |
| Print graph ir. More... | |
| Graph | nnvm::pass::OrderMutation (Graph src) |
| Add control flow dependencies between nodes. More... | |
| Graph | nnvm::pass::InferShape (Graph graph, ShapeVector shape_inputs, std::string shape_attr_key="") |
| Infer shapes in the graph given the information. More... | |
| Graph | nnvm::pass::InferType (Graph graph, DTypeVector dtype_inputs, std::string dtype_attr_key="") |
| Infer types in the graph given the information. More... | |
| Graph | nnvm::pass::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. More... | |
| Graph | nnvm::pass::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. More... | |
Detailed Description
Pass functions that simply redirect the calls to ApplyPass.
This file serves as documentation on how to use functions implemented in "src/pass". It is totally optional to add these functions when you add a new pass, since ApplyPass can be directly called.
