caffe2.python.layers.sparse_lookup.SparseLookup Class Reference

Inheritance diagram for caffe2.python.layers.sparse_lookup.SparseLookup:

Public Member Functions
def	__init__ (self, model, input_record, inner_shape, reducer, weight_init=None, weight_optim=None, name='sparse_lookup', regularizer=None, kwargs)
def	get_memory_usage (self)
def	get_fp16_compatible_parameters (self)
def	support_8bit (self)
def	get_8bits_compatible_parameters (self, fused=True)
def	add_train_ops (self, net)
def	add_ops (self, net)

Public Attributes
	external_weights
	reducer
	input_dim
	shape
	weight_init
	w
	scale_bias_init
	scale_bias
	output_schema
	trainer_version

Detailed Description

Definition at line 47 of file sparse_lookup.py.

---

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

• caffe2/python/layers/sparse_lookup.py