caffe_translator.TranslatorRegistry Class Reference

Inheritance diagram for caffe_translator.TranslatorRegistry:

Public Member Functions
def	Register (cls, op_name)
def	TranslateLayer (cls, layer, pretrained_blobs, is_test)
def	TranslateModel (cls, caffe_net, pretrained_net, is_test=False, net_state=None)

Static Public Attributes
dictionary	registry_ = {}

Detailed Description

Definition at line 50 of file caffe_translator.py.

Member Function Documentation

Register()

def caffe_translator.TranslatorRegistry.Register	(		cls,
			op_name
	)

A decorator for registering gradient mappings.

Definition at line 54 of file caffe_translator.py.

---

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

• caffe2/python/caffe_translator.py