Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
caffe2::MPSCNNContext Struct Reference

Public Member Functions

id< MTLComputePipelineState > getPipelineState (NSString *kernel)
 
id< MTLComputePipelineState > getSpecializedPipelineState (NSString *kernel, const std::vector< ushort > &constants)
 

Data Fields

id< MTLDevice > device
 
id< MTLCommandQueue > commandQueue
 
id< MTLLibrary > library
 

Detailed Description

Definition at line 16 of file mpscnn_context.h.


The documentation for this struct was generated from the following file: