Class XlaRecvTPUEmbeddingDeduplicationData.Inputs

java.lang.Object

org.tensorflow.op.RawOpInputs<XlaRecvTPUEmbeddingDeduplicationData>

org.tensorflow.op.xla.XlaRecvTPUEmbeddingDeduplicationData.Inputs

Enclosing class:

XlaRecvTPUEmbeddingDeduplicationData

public static class XlaRecvTPUEmbeddingDeduplicationData.Inputs
extends RawOpInputs<XlaRecvTPUEmbeddingDeduplicationData>

Field Summary

Fields

Modifier and Type	Field	Description
final String	config	Serialized TPUEmbeddingConfiguration proto.
final String	embeddingPartitions	Serialized EmbeddingPartitionsProto proto.
final String	hbmBuffersConfig	Serialized HbmBuffersConfig proto.
final String	tpuTopology	Serialized TpuTopologyArgsProto proto.

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)
XlaRecvTPUEmbeddingDeduplicationData	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

config

public final String config

Serialized TPUEmbeddingConfiguration proto.

embeddingPartitions

public final String embeddingPartitions

Serialized EmbeddingPartitionsProto proto.

hbmBuffersConfig

public final String hbmBuffersConfig

Serialized HbmBuffersConfig proto.

tpuTopology

public final String tpuTopology

Serialized TpuTopologyArgsProto proto.

Constructor Details

Inputs

public Inputs(GraphOperation op)