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:

• caffe2/python/layers/bucket_weighted.py