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 196 of file

Member Function Documentation

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

Definition at line 200 of file

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