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 get_8bits_compatible_parameters (self, fused=True)
 
def add_ops (self, net)
 

Public Attributes

 external_weights
 
 reducer
 
 shape
 
 weight_init
 
 w
 
 scale_bias_init
 
 scale_bias
 
 output_schema
 

Detailed Description

Definition at line 56 of file sparse_lookup.py.


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