Searched refs:_partition_graph_module (Results 1 – 3 of 3) sorted by relevance
101 def _partition_graph_module( member in Backend2PartitionerDemo110 submodule_partition_tags = self._partition_graph_module(submodule)121 partition_tags = self._partition_graph_module(exported_program.graph_module)178 def _partition_graph_module( member in Backend1PartitionerDemo190 self._partition_graph_module(submodule)207 partition_tags = self._partition_graph_module(exported_program.graph_module)
67 def _partition_graph_module( member in AddMulPartitionerDemo82 ret_partition_tags = self._partition_graph_module(submodule)88 partition_tags = self._partition_graph_module(exported_program.graph_module)245 def _partition_graph_module( member in NonDecompTestPartitioner258 ret_partition_tags = self._partition_graph_module(submodule)263 partition_tags = self._partition_graph_module(exported_program.graph_module)
46 def _partition_graph_module( member in ExamplePartitioner79 submodule_partition_tags = self._partition_graph_module(submodule)85 partition_tag = self._partition_graph_module(exported_program.graph_module)