Searched refs:module_class (Results 1 – 10 of 10) sorted by relevance
| /external/executorch/test/models/ |
| D | export_program.py | 191 module_class: Type[nn.Module],
198 if hasattr(module_class, "get_export_kwargs"):
200 export_kwargs = module_class.get_export_kwargs()
202 if hasattr(module_class, "export_joint"):
203 export_joint = module_class.export_joint() # pyre-ignore
204 if hasattr(module_class, "get_method_names_to_export"):
206 methods = module_class.get_method_names_to_export()
210 module_class,
246 module_class = getattr(sys.modules[__name__], module, None)
247 if not (inspect.isclass(module_class) and issubclass(module_class, nn.Module)):
[all …]
|
| D | export_delegated_program.py | 75 module_class: Type[nn.Module],
83 eager_module = module_class().eval()
165 module_class = getattr(sys.modules[__name__], module, None)
166 if not (inspect.isclass(module_class) and issubclass(module_class, nn.Module)):
168 module_names_to_classes[module] = module_class
172 for module_name, module_class in module_names_to_classes.items():
184 module_class,
|
| /external/pigweed/pw_cli/py/pw_cli/ |
| D | allowed_caller.py | 36 module_class = None
38 module_class = self_obj.__class__.__name__
43 self_class=module_class,
|
| /external/pytorch/torch/onnx/_internal/fx/passes/ |
| D | modularization.py | 50 module_class: type | str | None,
54 self._module_class = module_class
134 module_name, module_class = raw_meta
135 return _ModuleMeta(module_name, module_class, raw_meta)
142 module_name, (qualified_name, module_class) = raw_meta
143 return _ModuleMeta(module_name, module_class, raw_meta)
316 def module_class(self) -> type | str | None: member in _ModuleStackMeta
690 if (module_class := self._stack_meta.module_class) is not None:
692 _pass.PackageInfo.from_python_class(module_class)
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | custom_config.py | 102 module_class: Type,
115 self.standalone_module_classes[module_class] = StandaloneModuleConfigEntry(
273 module_class,
289 module_class,
346 for module_class, sm_config_entry in self.standalone_module_classes.items():
350 _make_tuple(module_class, sm_config_entry)
|
| /external/executorch/test/end2end/ |
| D | exported_module.py | 65 module_class: Type[nn.Module],
123 eager_module = module_class().eval()
|
| D | test_end2end.py | 509 module_class=module_cls,
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_fsdp.py | 652 module_class: Type[FSDPTestModel],
672 module_class.init(*model_args, **model_kwargs),
|
| /external/pytorch/test/nn/ |
| D | test_pooling.py | 765 module_class = getattr(torch.nn, module_name)
766 unpool = module_class(module_size).to(device)
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_onnxruntime.py | 784 def test_atleast_nd_list_input(self, module_class: torch.nn.Module):
792 self.run_test(module_class(), inputs)
810 self, module_class: torch.nn.Module, inputs: torch.Tensor
812 self.run_test(module_class(), inputs)
13651 name_fn=lambda module_class: module_class.__name__,
13656 def test_optional_output(self, module_class: Type[torch.nn.Module], x_size: int):
13659 model = torch.jit.script(module_class())
13686 module_class(),
|