 |
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
|
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| CudnnConvGradientOp (const OperatorDef &operator_def, Workspace *ws) | |
| template<typename T_X , typename T_DY , typename T_W , typename T_B , typename T_DX , typename T_DW , typename T_DB > | |
| bool | DoRunWithType () |
| bool | RunOnDevice () override |
 Public Member Functions inherited from caffe2::CudnnConvOpBase | |
| CudnnConvOpBase (const OperatorDef &operator_def, Workspace *ws) | |
| Public Member Functions inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| ConvPoolOpBase (const OperatorDef &operator_def, Workspace *ws) | |
| vector< int > | GetDims (const Tensor &input) |
| int | GetDimsSize (const Tensor &input) |
| std::vector< int64_t > | GetOutputSize (const Tensor &input, int output_channel) |
| void | SetOutputSize (const Tensor &input, Tensor *output, int output_channel) |
| void | ComputePads (const vector< int > &dims) |
| bool | HasPad () const |
| bool | HasStride () const |
| void | SetDeviceTensor (const std::vector< int > &data, Tensor *tensor) |
| void | SetBiasMultiplier (const int size, Tensor *bias_multiplier_) |
| bool | RunOnDevice () override |
| virtual bool | RunOnDeviceWithOrderNHWC () |
| virtual bool | RunOnDeviceWithOrderNCHW () |
| Public Member Functions inherited from caffe2::Operator< CUDAContext > | |
| Operator (const OperatorDef &operator_def, Workspace *ws) | |
| Operator (const c10::FunctionSchema &fn_schema, std::vector< c10::IValue > inputs, std::vector< at::Tensor > outputs) | |
| const Tensor & | Input (int idx, DeviceType type=CUDAContext::GetDeviceType()) |
| Retrieve a non-owning reference to the input at position 'idx' for this operator. More... | |
| Tensor | XOutput (int idx, at::IntArrayRef dims, at::TensorOptions options) |
| XOutput is a modernized version of Output which returns a Tensor rather than a Tensor* (the raw pointer in the latter case is useless, as Tensor is a pointer type.) | |
| Public Member Functions inherited from caffe2::OperatorBase | |
| OperatorBase (const OperatorDef &operator_def, Workspace *ws) | |
| OperatorBase (const c10::FunctionSchema &schema, std::vector< c10::IValue > inputs, std::vector< at::Tensor > outputs) | |
| bool | isLegacyOperator () const |
| Return true if the operator was instantiated with OperatorDef New operators should be instantiated with FunctionSchema. | |
| const c10::FunctionSchema & | getFunctionSchema () const |
| bool | HasArgument (const string &name) const |
| Checks if the operator has an argument of the given name. | |
| template<typename T > | |
| T | GetSingleArgument (const string &name, const T &default_value) const |
| template<typename T > | |
| bool | HasSingleArgumentOfType (const string &name) const |
| template<typename T > | |
| vector< T > | GetVectorFromIValueList (const c10::IValue &value) const |
| template<typename T > | |
| vector< T > | GetRepeatedArgument (const string &name, const vector< T > &default_value={}) const |
| template<typename T > | |
| const T & | Input (int idx) |
| template<typename T > | |
| const T & | Input (int idx, DeviceType type) |
| template<typename T > | |
| T * | Output (int idx) |
| template<typename T > | |
| T * | Output (int idx, DeviceType type) |
| Tensor | XOutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options) |
| void | SetOutputTensor (int idx, Tensor tensor) |
| Tensor | OutputTensorOrUndefined (int idx) |
| Tensor * | OutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options) |
| Tensor * | OutputTensorCopyFrom (int idx, at::TensorOptions options, const Tensor &src, bool async=false) |
| Tensor * | OutputTensorAlias (int idx, const Tensor &src) |
| template<typename T > | |
| T * | Output (int idx, T *allocated) |
| const Blob & | InputBlob (int idx) |
| Blob * | OutputBlob (int idx) |
| bool | IsInputOutputAlias (int i, int j) |
| template<typename T > | |
| bool | InputIsType (int idx) |
| bool | InputIsTensorType (int idx, DeviceType device_type) |
| template<typename T > | |
| bool | OutputIsType (int idx) |
| bool | OutputIsTensorType (int idx, DeviceType type) |
| int | InputSize () const |
| int | OutputSize () const |
| const vector< const Blob * > & | Inputs () const |
| const vector< Blob * > & | Outputs () |
| vector< TensorShape > | InputTensorShapes () const |
| virtual void | WaitEvent (const Event &ev, int=-1) |
| void | Wait (const OperatorBase &other, int stream_id=-1) |
| virtual void | WaitEvents (const std::vector< const Event * > &events, int=-1) |
| virtual void | Finish () |
| virtual bool | Run (int=0) |
| virtual bool | HasAsyncPart () const |
| virtual bool | SupportsAsyncScheduling () const |
| virtual bool | RunAsync (int stream_id=0) |
| virtual void | AddRelatedBlobInfo (EnforceNotMet *err) |
| const OperatorDef & | debug_def () const |
| void | set_debug_def (const std::shared_ptr< const OperatorDef > &operator_def) |
| bool | has_debug_def () const |
| void | RecordLastFailedOpNetPosition () |
| int | net_position () const |
| void | set_net_position (int idx) |
| const DeviceOption & | device_option () const |
| const Event & | event () const |
| Event & | event () |
| void | ResetEvent () |
| void | DisableEvent () |
| bool | IsEventDisabled () const |
| virtual void | SyncDeviceBarrierForObservers () |
| virtual bool | IsStreamFree (int) const |
| const std::string & | type () const |
| void | annotate_engine (const std::string &engine) |
| const std::string & | engine () const |
| void | SetExecutorHelper (ExecutorHelper *helper) |
| ExecutorHelper * | GetExecutorHelper () const |
| std::vector< at::Tensor > | move_newstyle_outputs ()&& |
| template<> | |
| NetDef | GetSingleArgument (const std::string &name, const NetDef &default_value) const |
| template<> | |
| vector< int > | GetVectorFromIValueList (const c10::IValue &value) const |
| template<> | |
| vector< float > | GetVectorFromIValueList (const c10::IValue &value) const |
| template<> | |
| vector< string > | GetVectorFromIValueList (const c10::IValue &value) const |
| Public Member Functions inherited from caffe2::Observable< OperatorBase > | |
| Observable (Observable &&)=default | |
| Observable & | operator= (Observable &&)=default |
| C10_DISABLE_COPY_AND_ASSIGN (Observable) | |
| const Observer * | AttachObserver (std::unique_ptr< Observer > observer) |
| std::unique_ptr< Observer > | DetachObserver (const Observer *observer_ptr) |
| Returns a unique_ptr to the removed observer. More... | |
| virtual size_t | NumObservers () |
| void | StartAllObservers () |
| void | StopAllObservers () |
Additional Inherited Members | |
| Public Types inherited from caffe2::Observable< OperatorBase > | |
| using | Observer = ObserverBase< OperatorBase > |
| Static Public Member Functions inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| static void | InferOutputSize (const at::IntArrayRef &input_dims, const int output_channel, const StorageOrder order, const bool global_pooling, const LegacyPadding legacy_pad, const std::vector< int > &dilation, const std::vector< int > &stride, std::vector< int > *kernel, std::vector< int > *pads, std::vector< int > *output_dims) |
| static void | InferOutputSize64 (const at::IntList &input_dims, const int output_channel, const StorageOrder order, const bool global_pooling, const LegacyPadding legacy_pad, const std::vector< int > &dilation, const std::vector< int > &stride, std::vector< int > *kernel, std::vector< int > *pads, std::vector< int64_t > *output_dims) |
| static struct OpSchema::Cost | CostInferenceForConv (const OperatorDef &def, const vector< TensorShape > &inputs) |
| static vector< TensorShape > | TensorInferenceForSchema (const OperatorDef &def, const vector< TensorShape > &in, int output_channel) |
| static std::vector< TensorShape > | TensorInferenceForConv (const OperatorDef &def, const std::vector< TensorShape > &in) |
| static std::vector< TensorShape > | TensorInferenceForPool (const OperatorDef &def, const std::vector< TensorShape > &in) |
| static std::vector< TensorShape > | TensorInferenceForLC (const OperatorDef &def, const std::vector< TensorShape > &in) |
| Data Fields inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| USE_OPERATOR_CONTEXT_FUNCTIONS | |
| Static Public Attributes inherited from caffe2::OperatorBase | |
| static const int | kNoNetPositionSet = -1 |
| Protected Member Functions inherited from caffe2::CudnnConvOpBase | |
| template<typename T > | |
| void | SetTensorNdDescriptorWithGroup (int size, cudnnTensorDescriptor_t tensorDesc, int N, int C, int H, int W, int D) |
| void | DuplicateConvDesc (cudnnConvolutionDescriptor_t input, size_t kernelDims, size_t dilationDims, cudnnConvolutionDescriptor_t copy) |
| template<typename T > | |
| cudnnDataType_t | DetermineComputeTypeFromInput (const T &X) |
| void | SetConvDescFromArguments () |
| void | SetConvDescComputeType (cudnnConvolutionDescriptor_t conv_desc, cudnnDataType_t math) |
| Protected Member Functions inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| int | pad_t () const |
| int | pad_l () const |
| int | pad_b () const |
| int | pad_r () const |
| int | kernel_h () const |
| int | kernel_w () const |
| int | stride_h () const |
| int | stride_w () const |
| int | dilation_h () const |
| int | dilation_w () const |
| Protected Member Functions inherited from caffe2::OperatorBase | |
| virtual void | RecordEvent (const char *=nullptr) |
| void | SetEventFinished (const char *err_msg=nullptr) |
| void | SetEventFinishedWithException (const char *err_msg=nullptr) |
| std::string | getErrorMsg () |
| C10_DISABLE_COPY_AND_ASSIGN (OperatorBase) | |
| Static Protected Member Functions inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| static void | ComputeSizeAndPad (const int in_size, const int stride, const int kernel, const int dilation, LegacyPadding legacy_pad, int *pad_head, int *pad_tail, int *out_size) |
| static void | ComputeSizeAndPad64 (const int in_size, const int stride, const int kernel, const int dilation, LegacyPadding legacy_pad, int *pad_head, int *pad_tail, int64_t *out_size) |
| Protected Attributes inherited from caffe2::CudnnConvOpBase | |
| vector< int64_t > | cudnn_input_dims_ |
| vector< int64_t > | cudnn_filter_dims_ |
| CuDNNWrapper | cudnn_wrapper_ |
| cudnnTensorDescriptor_t | bottom_desc_ |
| cudnnFilterDescriptor_t | filter_desc_ |
| cudnnTensorDescriptor_t | bias_desc_ |
| cudnnTensorDescriptor_t | top_desc_ |
| cudnnTensorDescriptor_t | top_desc_for_bias_ |
| cudnnConvolutionDescriptor_t | conv_desc_ |
| const size_t | cudnn_ws_nbytes_limit_ |
| size_t | cudnn_ws_nbytes_ |
| bool | exhaustive_search_ |
| bool | deterministic_ |
| size_t | cudnn_state_ |
| vector< int > | force_algo_ |
| bool | enable_tensor_core_ |
| cudnnDataType_t | compute_type_ |
| Protected Attributes inherited from caffe2::ConvPoolOpBase< CUDAContext > | |
| LegacyPadding | legacy_pad_ |
| bool | global_pooling_ |
| vector< int > | kernel_ |
| vector< int > | dilation_ |
| vector< int > | stride_ |
| vector< int > | pads_ |
| bool | float16_compute_ |
| int | group_ |
| StorageOrder | order_ |
| bool | shared_buffer_ |
| Workspace * | ws_ |
| Protected Attributes inherited from caffe2::OperatorBase | |
| std::unique_ptr< Event > | event_ |
| Protected Attributes inherited from caffe2::Observable< OperatorBase > | |
| std::vector< std::unique_ptr< Observer > > | observers_list_ |
Definition at line 454 of file conv_op_cudnn.cc.
1.8.11
