Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Attributes | List of all members
caffe2.python.caffe_translator.TranslatorRegistry Class Reference
Inheritance diagram for caffe2.python.caffe_translator.TranslatorRegistry:

Public Member Functions

def Register (cls, op_name)
 
def TranslateLayer (cls, layer, pretrained_blobs, is_test, kwargs)
 
def TranslateModel (cls, caffe_net, pretrained_net, is_test=False, net_state=None, remove_legacy_pad=False, input_dims=None)
 

Static Public Attributes

dictionary registry_ = {}
 

Detailed Description

Definition at line 211 of file caffe_translator.py.

Member Function Documentation

def caffe2.python.caffe_translator.TranslatorRegistry.Register (   cls,
  op_name 
)
A decorator for registering gradient mappings.

Definition at line 215 of file caffe_translator.py.


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