Home
last modified time | relevance | path

Searched refs:split_module (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Dnet_min_base.py298 split_module = split_by_tags(self.module, ["main_0", "minimize", "main_1"])
302 for child_name, _ in split_module.named_children(): # type: ignore[union-attr]
319 return split_module, submodule_name # type: ignore[return-value]
323 split_module: torch.fx.GraphModule,
338 submodule = getattr(split_module, submod_name)
339 a_input, b_input = self._get_submod_inputs(split_module, submod_name)
427 split_module, submod_name = self._build_submodule(cur_nodes)
428 self._run_and_compare(split_module, submod_name, [output_node_name])
504 split_module, submod_name = self._build_submodule(cur_nodes)
505 self._run_and_compare(split_module, submod_name, [node.name])
[all …]
H A Dsplitter_base.py216 split_module: torch.fx.GraphModule
866 split_module = split_by_tags(self.module, self.tags, return_tuple=self._return_tuple)
871 return split_module # type: ignore[return-value]
883 split_module = self()
885 for name, mod in split_module.named_children():
897 …submodule_inputs = generate_inputs_for_submodules(split_module, self.sample_input, submodule_names)
898 return SplitResult(split_module, submodule_inputs, self.non_acc_submodule_name)
H A D__init__.py9 from . import split_module
H A Dsplit_module.py44 def split_module( function
/aosp_15_r20/external/google-breakpad/src/tools/mac/dump_syms/
H A Ddump_syms_tool.cc167 bool split_module = in Start() local
170 split_module ? options.dsymPath : options.srcPath; in Start()
191 if (split_module && primary_file == options.dsymPath) { in Start()
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dconst_fold.py7 from torch.fx.passes.split_module import split_module
199 split = split_module(mod_traced, module, mod_partition)
H A Daccelerator_partitioner.py20 from torch.fx.passes.split_module import split_module
569 module_with_submodules = split_module(
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx_experimental.py44 from torch.fx.passes.split_module import split_module
768 module_with_submodules = split_module(
815 split = split_module(traced, mod, split_callback)
840 split = split_module(traced, mod, split_callback)
853 module_with_submodules = split_module(traced, m, lambda node: 0)
873 split = split_module(traced, mtt, lambda node: 0)
1195 module_with_submodule = split_module(traced, mm, split_cb)
1231 module_with_submodules = split_module(
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Ddistributed.py514 split_gm = fx.passes.split_module.split_module(
/aosp_15_r20/external/pytorch/torch/_export/
H A Dutils.py491 from torch.fx.passes.split_module import split_module
500 new_gm = split_module(
/aosp_15_r20/external/pytorch/test/expect/
H A DTestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect13 torch.fx.passes.split_module.Partition []
H A DTestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect67 torch.fx.passes.split_module.split_module(m: torch.fx.graph_module.GraphModule, root_m: torch.nn.mo…
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A D_IR.py23 from torch.fx.passes.split_module import split_module
733 split = split_module(traced, mod, split_callback) # type: ignore[arg-type]
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd.py744 split_gm = torch.fx.passes.split_module.split_module(
/aosp_15_r20/external/pytorch/
H A D.lintrunner.toml1300 'torch/fx/passes/split_module.py',
/aosp_15_r20/external/pytorch/docs/source/
H A Dfx.rst1188 .. py:module:: torch.fx.passes.split_module