Searched refs:num_wts (Results 1 – 1 of 1) sorted by relevance
369 def _create_weights(cls, wt_init, num_wts, num_shards, name): argument392 if num_shards == 1 and len(wt_init) == num_wts:404 sizes = cls._shard_sizes(num_wts, num_shards)