| 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 to wrap a new NcclAllReduce operation to the graph.
|
Copyright © 2015–2019. All rights reserved.