Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
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


Detailed Description

Definition at line 47 of file

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