Package | Description |
---|---|
org.tensorflow.op.core |
Modifier and Type | Method and Description |
---|---|
static <T extends Number> |
NcclAllReduce.create(Scope scope,
Operand<T> input,
String reduction,
Long numDevices,
String sharedName)
Factory method to create a class wrapping a new NcclAllReduce operation.
|
static <T extends Number> |
NcclAllReduce.create(Scope scope,
Operand<T> input,
String reduction,
Long numDevices,
String sharedName)
Factory method to create a class wrapping a new NcclAllReduce operation.
|
Copyright © 2022. All rights reserved.