Class PartitionedCall.Inputs

java.lang.Object

org.tensorflow.op.RawOpInputs<PartitionedCall>

org.tensorflow.op.tpu.PartitionedCall.Inputs

Enclosing class:

PartitionedCall

public static class PartitionedCall.Inputs
extends RawOpInputs<PartitionedCall>

Field Summary

Fields

Modifier and Type	Field	Description
final Iterable<Operand<?>>	args	The arguments to the function.
final long	autotunerThresh	The autotunerThresh attribute
final Operand<TInt32>	deviceOrdinal	The TPU device ordinal to run the function on.
final DataType[]	Tin	The types of the arguments to the function.
final DataType[]	Tout	The types of the outputs of the function.

Constructor Summary

Constructors

Constructor	Description
Inputs(GraphOperation op)

Method Summary

Methods inherited from class RawOpInputs

attributeMetadata, attributeNames, attributes, attributeValue, attributeValues, equals, getOutputs, hashCode, toString

Methods inherited from class Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Field Details

args

public final Iterable<Operand<?>> args

The arguments to the function.

deviceOrdinal

public final Operand<TInt32> deviceOrdinal

The TPU device ordinal to run the function on.

Tin

public final DataType[] Tin

The types of the arguments to the function.

Tout

public final DataType[] Tout

The types of the outputs of the function.

autotunerThresh

public final long autotunerThresh

The autotunerThresh attribute

Constructor Details

Inputs

public Inputs(GraphOperation op)