Searched refs:LoadTPUEmbeddingCenteredRMSPropParametersAttr (Results 1 – 1 of 1) sorted by relevance
28554 type LoadTPUEmbeddingCenteredRMSPropParametersAttr func(optionalAttr) type28558 …mbeddingCenteredRMSPropParametersTableId(value int64) LoadTPUEmbeddingCenteredRMSPropParametersAtt…28566 …ddingCenteredRMSPropParametersTableName(value string) LoadTPUEmbeddingCenteredRMSPropParametersAtt…28574 …mbeddingCenteredRMSPropParametersConfig(value string) LoadTPUEmbeddingCenteredRMSPropParametersAtt…28597 …Output, num_shards int64, shard_id int64, optional ...LoadTPUEmbeddingCenteredRMSPropParametersAtt… argument