Searched refs:EnqueueTPUEmbeddingSparseBatchAttr (Results 1 – 1 of 1) sorted by relevance
14563 type EnqueueTPUEmbeddingSparseBatchAttr func(optionalAttr) type14570 func EnqueueTPUEmbeddingSparseBatchDeviceOrdinal(value int64) EnqueueTPUEmbeddingSparseBatchAttr {14585 func EnqueueTPUEmbeddingSparseBatchCombiners(value []string) EnqueueTPUEmbeddingSparseBatchAttr {14624 …_weights []tf.Output, mode_override tf.Output, optional ...EnqueueTPUEmbeddingSparseBatchAttr) (o … argument