FinalizeTPUEmbedding

public final class FinalizeTPUEmbedding

An op that finalizes the TPUEmbedding configuration.

Public Methods

static FinalizeTPUEmbedding

create(Scope scope, Operand<String> commonConfig, Operand<String> memoryConfig) Factory method to create a class wrapping a new FinalizeTPUEmbedding operation.

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

Public Methods

public static FinalizeTPUEmbedding create (Scope scope, Operand<String> commonConfig, Operand<String> memoryConfig)

Factory method to create a class wrapping a new FinalizeTPUEmbedding operation.

Parameters

scope	current scope
commonConfig	A string-encoded common configuration proto containing metadata about the TPUEmbedding partitioner output and the HBM size (in bytes) required for operation.
memoryConfig	A string-encoded memory config proto containing metadata about the memory allocations reserved for TPUEmbedding.

Returns

• a new instance of FinalizeTPUEmbedding