Caffe2IOSPredictor Class Reference

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

Static Public Member Functions
static Caffe2IOSPredictor *	NewCaffe2IOSPredictor (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 14 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 11 of file ios_caffe_predictor.cc.

---

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

• caffe2/mobile/contrib/ios/ios_caffe_predictor.h
• caffe2/mobile/contrib/ios/ios_caffe_predictor.cc