org.tensorflow.op.core

Class ConfigureTPUEmbedding

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.ConfigureTPUEmbedding

All Implemented Interfaces:

Op

public final class ConfigureTPUEmbedding
extends PrimitiveOp

Sets up TPUEmbedding in a distributed TPU system.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static ConfigureTPUEmbedding	create(Scope scope, String config) Factory method to create a class wrapping a new ConfigureTPUEmbedding operation.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

public static ConfigureTPUEmbedding create(Scope scope,
                                           String config)

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

Parameters:

scope - current scope

config - Serialized tensorflow.tpu.TPUEmbeddingConfiguration that describes the embedding lookups of the program.

Returns:

a new instance of ConfigureTPUEmbedding