Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Data Fields
Caffe2IOSPredictor Class Referencefinal

Public Member Functions

void run (const Tensor &inData, Tensor &outData, std::string &errorMessage)
 

Static Public Member Functions

static Caffe2IOSPredictorNewCaffe2IOSPredictor (const caffe2::NetDef &init_net, const caffe2::NetDef &predict_net, bool disableMultithreadProcessing, bool allowMetalOperators)
 Allow converting eligible operators to Metal GPU framework accelerated operators. More...
 

Data Fields

const bool usingMetalOperators
 

Detailed Description

Definition at line 30 of file ios_caffe_predictor.h.

Member Function Documentation

Caffe2IOSPredictor * Caffe2IOSPredictor::NewCaffe2IOSPredictor ( const caffe2::NetDef &  init_net,
const caffe2::NetDef &  predict_net,
bool  disableMultithreadProcessing,
bool  allowMetalOperators 
)
static

Allow converting eligible operators to Metal GPU framework accelerated operators.

Setting this flag to true doesn't gaurantee predictor will be using Metal operators; Client code must check usingMetalOperators flag to determine predictor is using them.

Definition at line 27 of file ios_caffe_predictor.cc.


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