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

Modifier and Type	Method	Description
AttributeMetadata	attributeMetadata(String name)	Get the metadata for an attribute
Set<String>	attributeNames()	Get the names of this op's attributes
OperationAttributeInspector	attributes()	Get an inspector for the operation's attributes.
AttrValue	attributeValue(String name)	Get the value of an attribute as an AttrValue proto.
Map<String, AttrValue>	attributeValues()	Get all attribute value protos
final boolean	equals(Object obj)
PartitionedCall	getOutputs()	The outputs of this operation.
final int	hashCode()
final String	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)