AggregateSmallTensorsPacker

Packer that concatenates small tensors together for reduction.

This packer aggregates values that occupy less than maxBytes memory space into groups with at most maxGroups values each.

Annotations: @throws( ... )
Exceptions thrown: InvalidArgumentException If maxBytes or maxGroups is less than 1.

Linear Supertypes

Packer[PackInformation], AnyRef, Any

Instance Constructors

new AggregateSmallTensorsPacker(maxBytes: Long, maxGroups: Int)

maxBytes
Largest tensor eligible for aggregation, in terms of the number of bytes it takes up.
maxGroups
Largest permitted aggregation of small tensors.

Attributes
protected
Exceptions thrown
InvalidArgumentException If maxBytes or maxGroups is less than 1.

Value Members

final def !=(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def ##(): Int

Definition Classes
AnyRef → Any
final def ==(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def asInstanceOf[T0]: T0

Definition Classes
Any
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
final def eq(arg0: AnyRef): Boolean

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

Definition Classes
AnyRef → Any
def finalize(): Unit

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( classOf[java.lang.Throwable] )
final def getClass(): Class[_]

Definition Classes
AnyRef → Any
def hashCode(): Int

Definition Classes
AnyRef → Any
final def isInstanceOf[T0]: Boolean

Definition Classes
Any
val maxBytes: Long

Largest tensor eligible for aggregation, in terms of the number of bytes it takes up.
val maxGroups: Int

Largest permitted aggregation of small tensors.
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
def pack(grouped: Seq[Seq[OutputLike]]): (Seq[Seq[OutputLike]], Option[PackInformation])

Packs the provided values.
Packs the provided values.
grouped
Grouped values (per device).
returns
Packed values, ready for reduction, along with information that is necessary for unpacking later on.

Definition Classes
AggregateSmallTensorsPacker → Packer
Annotations
@throws( ... )
Exceptions thrown
InvalidArgumentException If the provided grouped values are inconsistent in any way.
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def toString(): String

Definition Classes
AnyRef → Any
def unpack(packed: Seq[Seq[OutputLike]], packInformation: Option[PackInformation]): Seq[Seq[OutputLike]]

Reverses the packing performed by pack, on the provided packed values.
Reverses the packing performed by pack, on the provided packed values.
packed
Packed values to unpack.
packInformation
Information from the packing process that is necessary for unpacking.
returns
Unpacked packed.

Definition Classes
AggregateSmallTensorsPacker → Packer
Annotations
@throws( ... )
Exceptions thrown
InvalidArgumentException If not pack information is provided, while it is actually necessary.
final def wait(): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long, arg1: Int): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )

Related Docs: object AggregateSmallTensorsPacker | package packers

class AggregateSmallTensorsPacker extends Packer[PackInformation]

Instance Constructors

new AggregateSmallTensorsPacker(maxBytes: Long, maxGroups: Int)

Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

final def asInstanceOf[T0]: T0

def clone(): AnyRef

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def finalize(): Unit

final def getClass(): Class[_]

def hashCode(): Int

final def isInstanceOf[T0]: Boolean

val maxBytes: Long

val maxGroups: Int

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

def pack(grouped: Seq[Seq[OutputLike]]): (Seq[Seq[OutputLike]], Option[PackInformation])

final def synchronized[T0](arg0: ⇒ T0): T0

def toString(): String

def unpack(packed: Seq[Seq[OutputLike]], packInformation: Option[PackInformation]): Seq[Seq[OutputLike]]

final def wait(): Unit

final def wait(arg0: Long, arg1: Int): Unit

final def wait(arg0: Long): Unit

Inherited from Packer[PackInformation]

Inherited from AnyRef

Inherited from Any

Ungrouped