Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
layers.sparse_to_dense.SparseToDense Class Reference
Inheritance diagram for layers.sparse_to_dense.SparseToDense:
layers.layers.ModelLayer

Public Member Functions

def __init__ (self, model, input_record, input_specs, name='sparse_to_dense', kwargs)
 
def add_ops (self, net)
 
def get_metadata (self)
 
- Public Member Functions inherited from layers.layers.ModelLayer
def __init__ (self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs)
 
def get_type (self)
 
def predict_input_record (self)
 
def input_record (self)
 
def predict_output_schema (self)
 
def predict_output_schema (self, output_schema)
 
def output_schema (self)
 
def output_schema (self, output_schema)
 
def get_parameters (self)
 
def get_fp16_compatible_parameters (self)
 
def get_memory_usage (self)
 
def add_operators (self, net, init_net=None, context=InstantiationContext.TRAINING)
 
def add_ops (self, net)
 
def add_eval_ops (self, net)
 
def add_train_ops (self, net)
 
def add_calibration_ops (self, net)
 

Public Attributes

 input_specs
 
 output_schema
 
 zero
 
 zero_range
 
- Public Attributes inherited from layers.layers.ModelLayer
 name
 
 model
 
 kwargs
 
 request_only
 
 eval_output_schema
 
 tags
 
 params
 

Detailed Description

Definition at line 15 of file sparse_to_dense.py.

Constructor & Destructor Documentation

◆ __init__()

def layers.sparse_to_dense.SparseToDense.__init__ (   self,
  model,
  input_record,
  input_specs,
  name = 'sparse_to_dense',
  kwargs 
)
`input_specs` follows the format of FeatureSpec from schema. To be more
precise it's a namedtuple that should have:
    'feature_type', 'feature_names', 'feature_ids'

Definition at line 19 of file sparse_to_dense.py.


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