Searched defs:local_module (Results 1 – 1 of 1) sorted by relevance
39 local_module, argument68 self, local_module, dist_module, inp_size, rank0_only=True, rowwise=False argument