Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::onnx::OnnxExporter Class Reference

Public Member Functions

 OnnxExporter (DummyName *dummy=nullptr)
 
ConvertedResult Caffe2OpToOnnxNodes (const caffe2::OperatorDef &def, const std::unordered_map< std::string, caffe2::TensorShape > &shapes)
 
void InitOpToTensorProto (const caffe2::OperatorDef &def, TensorProto *tensor)
 

Detailed Description

Definition at line 36 of file onnx_exporter.h.


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