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 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 200 of file caffe_translator.py.

---

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

• caffe2/python/caffe_translator.py