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

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