Searched refs:LoadTPUEmbeddingMomentumParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
29118 type LoadTPUEmbeddingMomentumParametersGradAccumDebugAttr func(optionalAttr) type29124 …entumParametersGradAccumDebugTableId(value int64) LoadTPUEmbeddingMomentumParametersGradAccumDebug…29132 …umParametersGradAccumDebugTableName(value string) LoadTPUEmbeddingMomentumParametersGradAccumDebug…29154 …ut, num_shards int64, shard_id int64, optional ...LoadTPUEmbeddingMomentumParametersGradAccumDebug… argument