Searched refs:_get_submodule_n_params (Results 1 – 1 of 1) sorted by relevance
14 def _get_submodule_n_params(module: nn.Module, path: str): function46 param_list.append((*_get_submodule_n_params(module, name), dtensor))66 param_list.append((*_get_submodule_n_params(module, name), t))