Inheritance diagram for caffe2::BackendTransformerBase:

Public Member Functions
const std::unordered_map< std::string, std::string > &	input_mapping () const

const std::unordered_map< std::string, std::string > &	reverse_input_mapping () const

virtual void	transform (Workspace ws, NetDef pred_net, const std::vector< std::string > &weight_names, const std::unordered_map< std::string, TensorShape > &shape_hints, const std::unordered_set< int > &blacklisted_ops)=0

Protected Member Functions
std::string	getModelId (const NetDef &net)

std::unordered_map< std::string, TensorShape >	ssaRewriteAndMapNames (Workspace ws, NetDef pred_net, const std::unordered_map< std::string, TensorShape > &input_shape_hints)

TensorProto	wrapShapeInfoIntoTensorProto (const std::string &name, const ShapeInfo &shape_info) const

ShapeInfoMap	inferShapes (Workspace ws, NetDef pred_net, const std::unordered_map< std::string, TensorShape > &shape_hints_mapped, const BoundShapeSpec &spec)

Protected Attributes
std::unordered_map< std::string, std::string >	input_mapping_

std::unordered_map< std::string, std::string >	reverse_input_mapping_

Detailed Description

Definition at line 20 of file backend_transformer_base.h.

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

Public Member Functions