Home
last modified time | relevance | path

Searched defs:Module (Results 76 – 100 of 2267) sorted by relevance

12345678910>>...91

/aosp_15_r20/external/pytorch/torch/ao/pruning/sparsifier/
H A Dbase_sparsifier.py142 model: nn.Module,
300 module: nn.Module,
301 mapping: Optional[Dict[Type[nn.Module], Type[nn.Module]]] = None,
303 parameterization: Type[nn.Module] = FakeSparsity,
350 def update_mask(self, module: nn.Module, tensor_name: str, **kwargs):
/aosp_15_r20/external/boringssl/src/util/fipstools/acvp/acvptool/acvp/
H A Dacvp.go595 type Module struct { struct
596 URL string `json:"url,omitempty"`
597 Name string `json:"name,omitempty"`
598 Version string `json:"version,omitempty"`
599 Type string `json:"type,omitempty"`
600 Website string `json:"website,omitempty"`
601 VendorURL string `json:"vendorUrl,omitempty"`
602 AddressURL string `json:"addressUrl,omitempty"`
603 ContactURLs []string `json:"contactUrls,omitempty"`
604 Description string `json:"description,omitempty"`
/aosp_15_r20/external/cronet/third_party/boringssl/src/util/fipstools/acvp/acvptool/acvp/
H A Dacvp.go595 type Module struct { struct
596 URL string `json:"url,omitempty"`
597 Name string `json:"name,omitempty"`
598 Version string `json:"version,omitempty"`
599 Type string `json:"type,omitempty"`
600 Website string `json:"website,omitempty"`
601 VendorURL string `json:"vendorUrl,omitempty"`
602 AddressURL string `json:"addressUrl,omitempty"`
603 ContactURLs []string `json:"contactUrls,omitempty"`
604 Description string `json:"description,omitempty"`
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Dreplicate.py29 def _is_script_module(module: Module) -> bool:
35 def _is_script_method(module: Module) -> bool:
60 def _replicatable_module(module: Module, memo: Optional[Set[Module]] = None) -> bool:
62 def descendant_modules(module: Module) -> Iterator[Module]:
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dmodule.py83 def __init__(self, hook: Callable, module: Optional["Module"] = None):
289 hook: Callable[["Module", _grad_t, _grad_t], Union[None, _grad_t]],
318 hook: Callable[["Module", _grad_t], Union[None, _grad_t]],
345 hook: Callable[["Module", _grad_t, _grad_t], Union[None, _grad_t]],
399 class Module: class
631 def add_module(self, name: str, module: Optional["Module"]) -> None:
659 def register_module(self, name: str, module: Optional["Module"]) -> None:
728 def set_submodule(self, target: str, module: "Module") -> None:
992 def apply(self: T, fn: Callable[["Module"], None]) -> T:
1344 hook: Callable[["Module", _grad_t], Union[None, _grad_t]],
[all …]
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dconvert.py93 modules: Dict[str, torch.nn.Module],
353 modules: Dict[str, torch.nn.Module],
511 def _is_conversion_supported(activation_post_process: torch.nn.Module) -> bool:
648 node: Node, modules: Dict[str, torch.nn.Module]
664 modules: Dict[str, torch.nn.Module],
728 modules: Dict[str, torch.nn.Module],
894 modules: Dict[str, torch.nn.Module],
/aosp_15_r20/build/soong/java/
H A Dclasspath_element.go31 Module() android.Module methodSpec
43 func (b *ClasspathFragmentElement) Module() android.Module { func
62 func (b *ClasspathLibraryElement) Module() android.Module { func
119 …teClasspathElements(ctx ClasspathElementContext, libraries []android.Module, fragments []android.M…
/aosp_15_r20/external/llvm/tools/bugpoint/
H A DMiscompilation.cpp185 bool (*F)(BugDriver &, std::unique_ptr<Module>, in ReduceMiscompilingFunctions()
298 bool (*TestFn)(BugDriver &, std::unique_ptr<Module>, in ExtractLoops()
446 bool (*F)(BugDriver &, std::unique_ptr<Module>, in ReduceMiscompiledBlocks()
529 bool (*TestFn)(BugDriver &, std::unique_ptr<Module>, in ExtractBlocks()
610 bool (*TestFn)(BugDriver &, std::unique_ptr<Module>, in DebugAMiscompilation()
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantize_fx.py24 model: Union[GraphModule, torch.nn.Module],
35 def _check_is_graph_module(model: torch.nn.Module) -> None:
59 def _swap_ff_with_fxff(model: torch.nn.Module) -> None:
91 model: torch.nn.Module,
163 model: torch.nn.Module,
205 model: torch.nn.Module,
253 model: torch.nn.Module,
404 model: torch.nn.Module,
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/
H A Dcheckpoint_wrapper.py71 module: nn.Module,
89 module: nn.Module,
123 mod: torch.nn.Module,
175 def offload_wrapper(module: torch.nn.Module) -> torch.nn.Module:
198 module: torch.nn.Module,
253 auto_wrap_policy: Optional[Callable[[nn.Module, bool, int], bool]] = None,
/aosp_15_r20/build/soong/android/
H A Dandroidmk.go498 ModuleDir(module blueprint.Module) string
499 ModuleSubDir(module blueprint.Module) string
501 otherModuleProvider(module blueprint.Module, provider blueprint.AnyProviderKey) (any, bool)
502 ModuleType(module blueprint.Module) string
503 OtherModulePropertyErrorf(module Module, property string, fmt string, args ...interface{})
739 …ngletonContext, absMkFile string, moduleInfoJSONPath WritablePath, mods []blueprint.Module) error {
799 …tx SingletonContext, w io.Writer, moduleInfoJSONs *[]*ModuleInfoJSON, mod blueprint.Module) error {
854 mod blueprint.Module, provider AndroidMkDataProvider) error {
949 mod blueprint.Module, provider AndroidMkEntriesProvider) error {
972 func ShouldSkipAndroidMkProcessing(ctx ConfigurableEvaluatorContext, module Module) bool {
[all …]
H A Dprebuilt.go344 func IsModulePreferred(module Module) bool {
358 func IsModulePrebuilt(module Module) bool {
365 func GetEmbeddedPrebuilt(module Module) *Prebuilt {
383 func PrebuiltGetPreferred(ctx BaseModuleContext, module Module) Module {
480 func checkInvariantsForSourceAndPrebuilt(ctx BaseModuleContext, s, p Module) {
551 …gedModules(ctx BottomUpMutatorContext, psi PrebuiltSelectionInfoMap, allModulesInFamily []Module) {
664 func isSelected(psi PrebuiltSelectionInfoMap, m Module) bool {
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A D_named_member_accessor.py12 def set_tensor(module: "torch.nn.Module", name: str, tensor: torch.Tensor) -> None:
30 module: "torch.nn.Module",
84 module: "torch.nn.Module",
86 submodule: "torch.nn.Module",
114 def __init__(self, module: "torch.nn.Module") -> None:
154 def swap_submodule(self, path: str, value: "torch.nn.Module") -> "torch.nn.Module":
H A Dstateless.py14 module: "torch.nn.Module",
100 module: "torch.nn.Module",
164 module: "torch.nn.Module",
187 module: "torch.nn.Module",
265 module: "torch.nn.Module",
/aosp_15_r20/external/autotest/metadata/utils/
H A Dcontrol_files.py63 def _extract_name(module: ast.Module) -> Optional[str]:
123 def _extract_suites(module: ast.Module) -> List[str]:
136 def _extract_main_package(path: str, module: ast.Module) -> Optional[str]:
145 def _extract_main_file(path: str, module: ast.Module) -> Optional[str]:
157 def _find_run_test_calls(module: ast.Module) -> List[ast.Call]:
/aosp_15_r20/prebuilts/go/linux-x86/src/runtime/debug/
Dmod.go62 type Module struct { struct
63 Path string // module path
64 Version string // module version
65 Sum string // checksum
66 Replace *Module // replaced by this module
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcompile.py41 model: Union[torch.nn.Module, Callable],
75 model: Union[torch.nn.Module, Callable],
81 loss_fn : Union[torch.nn.Module, Callable, None] = None,
120 model : Union[torch.nn.Module, Callable],
124 loss_fn : Union[torch.nn.Module, Callable, None] = None,
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_mean_dim.py78 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
98 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
120 module: torch.nn.Module,
147 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
166 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
187 module: torch.nn.Module,
H A Dtest_clone.py44 self, module: torch.nn.Module, test_data: torch.Tensor
62 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
84 module: torch.nn.Module,
100 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
107 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
H A Dtest_slice.py36 self, module: torch.nn.Module, test_data: torch.Tensor
55 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor], permute: bool
79 module: torch.nn.Module,
98 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
105 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
H A Dtest_tanh.py41 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
60 def _test_tanh_tosa_BI_pipeline(self, module: torch.nn.Module, test_data: Tuple):
82 module: torch.nn.Module,
103 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
110 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py59 expected_module: nn.Module,
61 module: nn.Module,
73 expected_module: nn.Module,
75 module: nn.Module,
86 expected_module: nn.Module,
88 module: nn.Module,
101 expected_module: nn.Module,
103 module: nn.Module,
124 module: nn.Module,
146 expected_module: nn.Module,
[all …]
/aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/
H A Dtest_fully_shard_mixed_precision.py151 self, input: Dict[nn.Module, torch.Tensor], fsdp_model: nn.Module
162 input: Dict[nn.Module, torch.Tensor],
163 fsdp_model: nn.Module,
179 fsdp_model: nn.Module,
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/
H A Dprepare.py316 model: torch.nn.Module,
317 named_modules: Dict[str, torch.nn.Module],
399 model: torch.nn.Module,
400 named_modules: Dict[str, torch.nn.Module],
448 model: torch.nn.Module,
449 named_modules: Dict[str, torch.nn.Module],
/aosp_15_r20/external/executorch/backends/vulkan/_passes/
H A Dint4_weight_only_quantizer.py102 module: torch.nn.Module,
110 linear_class: Type[torch.nn.Module] = VkWeightOnlyInt4Linear,
183 self, model: torch.nn.Module
239 def _convert_for_runtime(self, model: torch.nn.Module) -> torch.nn.Module:
252 self, model: torch.nn.Module, *args: Any, **kwargs: Any

12345678910>>...91