Package | Description |
---|---|
org.tensorflow.op.tpu |
Modifier and Type | Method and Description |
---|---|
static PartitionedCall |
PartitionedCall.create(Scope scope,
Iterable<Operand<?>> args,
Operand<TInt32> deviceOrdinal,
List<Class<? extends TType>> Tout,
ConcreteFunction f,
PartitionedCall.Options... options)
Factory method to create a class wrapping a new TPUPartitionedCall operation.
|
Copyright © 2015–2022. All rights reserved.