English
Bahasa Indonesia
Español – América Latina
Français
Italiano
Polski
Português – Brasil
Tiếng Việt
Türkçe
Русский
עברית
العربيّة
فارسی
हिंदी
বাংলা
ภาษาไทย
中文 – 简体
日本語
한국어

GitHub

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 ()

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2022-05-17 UTC.