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.NNExtension.NNExtension Class Reference
Inheritance diagram for tools.cwrap.plugins.NNExtension.NNExtension:

Public Member Functions

def __init__ (self, module_name)
 
def process_full_file (self, code, template_path)
 
def process_wrapper (self, code, declaration)
 
def declare_module_methods (self)
 
def get_type_unpack (self, arg, option)
 
def get_type_check (self, arg, option)
 
def get_wrapper_template (self, declaration)
 

Public Attributes

 module_name
 
 declarations
 

Static Public Attributes

 TYPE_UNPACK
 
 TYPE_CHECK
 
 WRAPPER_TEMPLATE
 
 TYPE_NAMES
 

Detailed Description

Definition at line 34 of file NNExtension.py.


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