Searched refs:TPUEmbedding (Results 1 – 5 of 5) sorted by relevance
27 summary: "An op that enqueues a list of input batch tensors to TPUEmbedding."
54 summary: "An op that enqueues TPUEmbedding input indices from a SparseTensor."
258 tpu_embedding_ = tpu_embedding.TPUEmbedding(
208 class TPUEmbedding(object): class
270 class TPUEmbedding(embeddings.Embedding): class281 return super(TPUEmbedding, self).build(input_shape)