Searched refs:EnqueueTPUEmbeddingRaggedTensorBatchAttr (Results 1 – 1 of 1) sorted by relevance
14457 type EnqueueTPUEmbeddingRaggedTensorBatchAttr func(optionalAttr) type14464 …eTPUEmbeddingRaggedTensorBatchDeviceOrdinal(value int64) EnqueueTPUEmbeddingRaggedTensorBatchAttr {14479 …ueTPUEmbeddingRaggedTensorBatchCombiners(value []string) EnqueueTPUEmbeddingRaggedTensorBatchAttr {14487 …eddingRaggedTensorBatchMaxSequenceLengths(value []int64) EnqueueTPUEmbeddingRaggedTensorBatchAttr {14495 …eTPUEmbeddingRaggedTensorBatchNumFeatures(value []int64) EnqueueTPUEmbeddingRaggedTensorBatchAttr {14544 …mode_override tf.Output, table_ids []int64, optional ...EnqueueTPUEmbeddingRaggedTensorBatchAttr) … argument