Caffe2 - C++ API
A deep learning, cross platform ML framework
Protected Member Functions
caffe2::ConvToNNPackTransform Class Reference
Inheritance diagram for caffe2::ConvToNNPackTransform:
caffe2::SingleOpTransform caffe2::Transform

Protected Member Functions

bool MatchOperator (const OperatorDef &op) override
void ReplaceOperator (OperatorDef *op) override
- Protected Member Functions inherited from caffe2::SingleOpTransform
bool PatternRule (const transform::Graph &g, const std::vector< int > &subgraph, int idx) override
 The PatternRule essentially answers: Given the current subgraph (ordered), should we append the new node at idx?
bool ValidatorRule (const transform::Graph &g, const std::vector< int > &subgraph) override
 The ValidatorRule essentially answers: Given a subgraph, can we accept it?
bool ReplaceRule (const std::vector< int > &subgraph, transform::Graph *g_ptr) override
 The ReplaceRule actually mutates the graph, and applies the transformation upon the subgraph.
- Protected Member Functions inherited from caffe2::Transform
void SetPatternMatchType (PatternMatchType type)

Additional Inherited Members

- Public Types inherited from caffe2::Transform
 Determines the type of subgraphs that PatternMatch will find. More...
- Public Member Functions inherited from caffe2::Transform
NetDef ApplyTo (const NetDef &orig_net_def)
 Apply a Transform onto a NetDef. More...
std::vector< std::vector< int > > PatternMatch (const transform::Graph &graph)
 Generates all matches (stored as ordered subgraphs) and returns them. More...
void ReplacePattern (const std::vector< std::vector< int >> &matches, transform::Graph *graph)
 Applies the replace rule onto each of the matches found.

Detailed Description

Definition at line 10 of file conv_to_nnpack_transform.h.

The documentation for this class was generated from the following file: