Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tools.cwrap.plugins.CuDNNPlugin.CuDNNPlugin Class Reference
Inheritance diagram for tools.cwrap.plugins.CuDNNPlugin.CuDNNPlugin:

Public Member Functions

def __init__ (self)
 
def get_type_unpack (self, arg, option)
 
def get_type_check (self, arg, option)
 
def get_assign_args (self, arguments)
 
def get_wrapper_template (self, declaration)
 
def get_return_wrapper (self, option)
 
def get_arg_accessor (self, arg, option)
 
def process_declarations (self, declarations)
 
def filter_unique_options (self, options)
 
def preprocessor_guard (self, code, condition)
 
def process_wrapper (self, code, declaration)
 
def process_all_call_arg (self, code, option)
 
def declare_methods (self)
 
def process_full_file (self, code, template_path)
 

Public Attributes

 declarations
 

Static Public Attributes

 TYPE_UNPACK
 
 INPUT_ARGUMENT_MAP
 
 TYPE_CHECK
 
 RETURN_WRAPPER
 
 METHODS_DECLARATION
 
 WRAPPER_TEMPLATE
 
 RELEASE_ARG
 
 TYPE_NAMES
 

Detailed Description

Definition at line 8 of file CuDNNPlugin.py.


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