Package | Description |
---|---|
org.tensorflow.op | |
org.tensorflow.op.tpu |
Modifier and Type | Method and Description |
---|---|
<T extends TType> |
TpuOps.partitionedInput(Iterable<Operand<T>> inputs,
PartitionedInput.Options... options)
An op that groups a list of partitioned inputs together.
|
Modifier and Type | Method and Description |
---|---|
static <T extends TType> |
PartitionedInput.create(Scope scope,
Iterable<Operand<T>> inputs,
PartitionedInput.Options... options)
Factory method to create a class wrapping a new TPUPartitionedInput operation.
|
Copyright © 2015–2022. All rights reserved.