Searched refs:RetrieveTPUEmbeddingRMSPropParametersAttr (Results 1 – 1 of 1) sorted by relevance
15549 type RetrieveTPUEmbeddingRMSPropParametersAttr func(optionalAttr) type15555 …trieveTPUEmbeddingRMSPropParametersTableId(value int64) RetrieveTPUEmbeddingRMSPropParametersAttr {15563 …eveTPUEmbeddingRMSPropParametersTableName(value string) RetrieveTPUEmbeddingRMSPropParametersAttr {15577 …e *Scope, num_shards int64, shard_id int64, optional ...RetrieveTPUEmbeddingRMSPropParametersAttr)… argument