ONE - On-device Neural Engine
Loading...
Searching...
No Matches
mir_caffe Namespace Reference

Data Structures

class  CaffeOpCreator
 

Enumerations

enum class  CaffeOpType {
  absVal , accuracy , argMax , batchNorm ,
  batchReindex , bias , BNLL , clip ,
  concat , contrastiveLoss , convolution , crop ,
  data , deconvolution , dropout , dummyData ,
  eltwise , ELU , embed , euclidianLoss ,
  exp , filter , flatten , HDF5Data ,
  HDF5Output , hingeLoss , im2Col , imageData ,
  infogainLoss , innerProduct , input , log ,
  LRN , LSTM , memoryData , multinomialLogisticLoss ,
  MVN , parameter , pooling , power ,
  PReLU , python , recurrent , reduction ,
  ReLU , reshape , RNN , scale ,
  sigmoidCrossEntropyLoss , sigmoid , silence , slice ,
  softmax , softmaxWithLoss , split , SPP ,
  tanh , threshold , tile , windowData
}
 

Functions

std::unique_ptr< mir::GraphimportModelFromBinaryFile (const std::string &filename)
 
std::unique_ptr< mir::GraphimportModelFromTextFile (const std::string &filename)
 
std::unique_ptr< mir::GraphloadModel (const std::string &filename)
 

Enumeration Type Documentation

◆ CaffeOpType

enum class mir_caffe::CaffeOpType
strong
Enumerator
absVal 
accuracy 
argMax 
batchNorm 
batchReindex 
bias 
BNLL 
clip 
concat 
contrastiveLoss 
convolution 
crop 
data 
deconvolution 
dropout 
dummyData 
eltwise 
ELU 
embed 
euclidianLoss 
exp 
filter 
flatten 
HDF5Data 
HDF5Output 
hingeLoss 
im2Col 
imageData 
infogainLoss 
innerProduct 
input 
log 
LRN 
LSTM 
memoryData 
multinomialLogisticLoss 
MVN 
parameter 
pooling 
power 
PReLU 
python 
recurrent 
reduction 
ReLU 
reshape 
RNN 
scale 
sigmoidCrossEntropyLoss 
sigmoid 
silence 
slice 
softmax 
softmaxWithLoss 
split 
SPP 
tanh 
threshold 
tile 
windowData 

Definition at line 23 of file caffe_op_types.h.

24{
25 absVal,
27 argMax,
30 bias,
31 BNLL,
32 clip,
33 concat,
36 crop,
37 data,
39 dropout,
41 eltwise,
42 ELU,
43 embed,
45 exp,
46 filter,
47 flatten,
51 im2Col,
55 input,
56 log,
57 LRN,
58 LSTM,
61 MVN,
63 pooling,
64 power,
65 PReLU,
66 python,
69 ReLU,
70 reshape,
71 RNN,
72 scale,
74 sigmoid,
75 silence,
76 slice,
77 softmax,
79 split,
80 SPP,
81 tanh,
83 tile,
85};

Function Documentation

◆ importModelFromBinaryFile()

std::unique_ptr< mir::Graph > mir_caffe::importModelFromBinaryFile ( const std::string &  filename)

Definition at line 422 of file caffe_importer.cpp.

423{
424 CaffeImporter importer;
425 return importer.importModelFromBinaryFile(filename);
426}

Referenced by loadModel(), and main().

◆ importModelFromTextFile()

std::unique_ptr< mir::Graph > mir_caffe::importModelFromTextFile ( const std::string &  filename)

Definition at line 428 of file caffe_importer.cpp.

429{
430 CaffeImporter importer;
431 return importer.importModelFromTextFile(filename);
432}

◆ loadModel()

std::unique_ptr< mir::Graph > mir_caffe::loadModel ( const std::string &  filename)

Definition at line 434 of file caffe_importer.cpp.

435{
436 return importModelFromBinaryFile(filename);
437}
std::unique_ptr< mir::Graph > importModelFromBinaryFile(const std::string &filename)

References importModelFromBinaryFile().

Referenced by main().