| /external/pytorch/torch/_inductor/ |
| D | loop_body.py | 332 def add_submodule(self, block, prefix): member in LoopBody
489 name = self.body.add_submodule(None, "masked_subblock")
505 name = self.body.add_submodule(shim, "scan")
538 self.body.add_submodule(set_indirect, f"set_{var}"),
|
| D | ir.py | 8004 def add_submodule(self, block, prefix): member in LoopBody
8118 name = self.body.add_submodule(shim, "masked_subblock")
8136 name = self.body.add_submodule(shim, "scan")
8167 self.body.add_submodule(set_indirect, f"set_{var}"),
|
| /external/pytorch/test/expect/ |
| D | TestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect | 5 torch.fx.graph_module.GraphModule ['add_submodule', 'code', 'delete_all_unused_submodules', 'delete…
|
| D | TestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect | 28 torch.fx.graph_module.GraphModule.add_submodule(self, target: str, m: torch.nn.modules.module.Modul…
|
| /external/executorch/exir/tests/ |
| D | test_pass_infra.py | 141 gm.add_submodule("foo", Foo())
|
| /external/pytorch/torch/fx/passes/utils/ |
| D | fuser_utils.py | 193 gm.add_submodule(submodule_name, sub_gm)
|
| /external/pytorch/torch/fx/ |
| D | subgraph_rewriter.py | 66 gm.add_submodule(node.target, new_attr)
|
| D | graph_module.py | 602 def add_submodule(self, target: str, m: torch.nn.Module) -> bool: member in GraphModule
|
| /external/pytorch/torch/ao/quantization/fx/_model_report/ |
| D | model_report.py | 226 self._model.add_submodule(obs_fqn, obs_to_insert)
|
| /external/pytorch/torch/_dynamo/backends/ |
| D | distributed.py | 276 self.module.add_submodule(n.target, compiled_submod_real)
|
| /external/python/cpython3/Modules/ |
| D | pyexpat.c | 1677 add_submodule(PyObject *mod, const char *fullname) in add_submodule() function
1825 PyObject *errors_module = add_submodule(mod, MODULE_NAME ".errors"); in add_errors_module()
1876 PyObject *model_module = add_submodule(mod, MODULE_NAME ".model"); in add_model_module()
|
| /external/pytorch/torch/onnx/_internal/fx/passes/ |
| D | modularization.py | 636 self._reference_module.add_submodule(unique_submodule_name, submodule)
|
| /external/pytorch/test/quantization/fx/ |
| D | test_model_report_fx.py | 839 model_prep.add_submodule(pre_obs_fqn, obs_to_insert)
846 model_prep.add_submodule(post_obs_fqn, obs_to_insert)
|
| /external/pytorch/test/ |
| D | test_fx.py | 3140 a.add_submodule("net_b.net_c.dropout", torch.nn.Dropout(p=0.2))
3235 a.add_submodule("net_b.embedding", torch.nn.Embedding(10, 3))
3236 a.add_submodule("net_b.net_c.embedding", torch.nn.Embedding(10, 3))
3237 a.add_submodule("net_b.net_c.rnn", torch.nn.RNN(10, 20, 2))
3238 a.add_submodule("batch_norm_2d", torch.nn.BatchNorm2d(100))
4247 gm.add_submodule("foo", Add())
|