CollectiveAssignGroupV2

public final class CollectiveAssignGroupV2

Assign group keys based on group assignment.

Public Methods

static CollectiveAssignGroupV2	create(Scope scope, Operand<Integer> groupAssignment, Operand<Integer> deviceIndex, Operand<Integer> baseKey) Factory method to create a class wrapping a new CollectiveAssignGroupV2 operation.
Output<Integer>	groupKey()
Output<Integer>	groupSize()

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

Public Methods

public static CollectiveAssignGroupV2 create (Scope scope, Operand<Integer> groupAssignment, Operand<Integer> deviceIndex, Operand<Integer> baseKey)

Factory method to create a class wrapping a new CollectiveAssignGroupV2 operation.

Parameters

scope	current scope

Returns

• a new instance of CollectiveAssignGroupV2

public Output<Integer> groupKey ()

public Output<Integer> groupSize ()