Package | Description |
---|---|
org.tensorflow.op.core |
Modifier and Type | Method and Description |
---|---|
static <T> AllToAll<T> |
AllToAll.create(Scope scope,
Operand<T> input,
Operand<Integer> groupAssignment,
Long concatDimension,
Long splitDimension,
Long splitCount)
Factory method to create a class wrapping a new AllToAll operation.
|
static <T> AllToAll<T> |
AllToAll.create(Scope scope,
Operand<T> input,
Operand<Integer> groupAssignment,
Long concatDimension,
Long splitDimension,
Long splitCount)
Factory method to create a class wrapping a new AllToAll operation.
|
Copyright © 2022. All rights reserved.