Inheritance diagram for caffe2::OnnxifiTransformer:

Public Member Functions
	OnnxifiTransformer (const OnnxifiTransformerOptions &opts)

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) override

Public Member Functions inherited from caffe2::BackendTransformerBase
const std::unordered_map< std::string, std::string > &	input_mapping () const

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

Additional Inherited Members
Protected Member Functions inherited from caffe2::BackendTransformerBase
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 inherited from caffe2::BackendTransformerBase
std::unordered_map< std::string, std::string >	input_mapping_

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

Detailed Description

Definition at line 42 of file onnxifi_transformer.h.

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

Public Member Functions