Searched refs:RetrieveTPUEmbeddingADAMParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
24980 type RetrieveTPUEmbeddingADAMParametersGradAccumDebugAttr func(optionalAttr) type24986 …gADAMParametersGradAccumDebugTableId(value int64) RetrieveTPUEmbeddingADAMParametersGradAccumDebug…24994 …AMParametersGradAccumDebugTableName(value string) RetrieveTPUEmbeddingADAMParametersGradAccumDebug…25008 …pe, num_shards int64, shard_id int64, optional ...RetrieveTPUEmbeddingADAMParametersGradAccumDebug… argument