Searched refs:RetrieveTPUEmbeddingMomentumParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
24682 type RetrieveTPUEmbeddingMomentumParametersGradAccumDebugAttr func(optionalAttr) type24688 …tumParametersGradAccumDebugTableId(value int64) RetrieveTPUEmbeddingMomentumParametersGradAccumDeb…24696 …ParametersGradAccumDebugTableName(value string) RetrieveTPUEmbeddingMomentumParametersGradAccumDeb…24710 …, num_shards int64, shard_id int64, optional ...RetrieveTPUEmbeddingMomentumParametersGradAccumDeb… argument