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

Public Member Functions

def __init__ (self, model, input_record, max_score=0, bucket_boundaries=None, weight_optim=None, name="bucket_weighted")
 
def get_memory_usage (self)
 
def add_ops (self, net)
 

Public Attributes

 bucket_boundaries
 
 shape
 
 bucket_w
 
 output_schema
 

Detailed Description

Definition at line 22 of file bucket_weighted.py.


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