Searched refs:RetrieveTPUEmbeddingRMSPropParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
26454 type RetrieveTPUEmbeddingRMSPropParametersGradAccumDebugAttr func(optionalAttr) type26460 …PropParametersGradAccumDebugTableId(value int64) RetrieveTPUEmbeddingRMSPropParametersGradAccumDeb…26468 …pParametersGradAccumDebugTableName(value string) RetrieveTPUEmbeddingRMSPropParametersGradAccumDeb…26482 …e, num_shards int64, shard_id int64, optional ...RetrieveTPUEmbeddingRMSPropParametersGradAccumDeb… argument