public static class PartitionedCall.Inputs extends RawOpInputs<PartitionedCall>
Modifier and Type | Field and Description |
---|---|
Iterable<Operand<?>> |
args
The arguments to the function.
|
long |
autotunerThresh
The autotunerThresh attribute
|
Operand<TInt32> |
deviceOrdinal
The TPU device ordinal to run the function on.
|
DataType[] |
Tin
The types of the arguments to the function.
|
DataType[] |
Tout
The types of the outputs of the function.
|
Constructor and Description |
---|
Inputs(GraphOperation op) |
attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString
public final Operand<TInt32> deviceOrdinal
public final DataType[] Tin
public final DataType[] Tout
public final long autotunerThresh
public Inputs(GraphOperation op)
Copyright © 2015–2022. All rights reserved.