Searched refs:LoadTPUEmbeddingADAMParametersAttr (Results 1 – 1 of 1) sorted by relevance
16924 type LoadTPUEmbeddingADAMParametersAttr func(optionalAttr) type16930 func LoadTPUEmbeddingADAMParametersTableId(value int64) LoadTPUEmbeddingADAMParametersAttr {16938 func LoadTPUEmbeddingADAMParametersTableName(value string) LoadTPUEmbeddingADAMParametersAttr {16960 …s tf.Output, num_shards int64, shard_id int64, optional ...LoadTPUEmbeddingADAMParametersAttr) (o … argument