Searched refs:mod_partition (Results 1 – 2 of 2) sorted by relevance
196 def mod_partition(node: torch.fx.Node): function199 split = split_module(mod_traced, module, mod_partition)
761 def mod_partition(node: Node): function769 my_module_traced, my_module, mod_partition