| /external/perfetto/src/trace_processor/sqlite/bindings/ |
| D | sqlite_module.h | 31 struct Module { struct
39 static constexpr enum { kEponymousOnly, kCreateOnly } kType = kCreateOnly;
45 static constexpr bool kSupportsWrites = true;
51 static constexpr bool kDoesOverloadFunctions = true;
56 using Context = void;
62 using Vtab = sqlite3_vtab;
68 using Cursor = sqlite3_vtab_cursor;
174 using FindFunctionFn = void(sqlite3_context*, int, sqlite3_value**);
198 static constexpr sqlite3_module CreateModule() { in CreateModule() argument
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | wrap.py | 44 root_module: nn.Module,
45 fn: Callable[[nn.Module], Optional[nn.Module]],
58 module: nn.Module,
60 parent_module: Optional[nn.Module],
85 root_module: nn.Module,
86 target_module_to_kwargs: Dict[nn.Module, Dict[str, Any]],
95 def fn(module: nn.Module) -> Optional[nn.Module]:
107 root_module: nn.Module,
108 module_classes: Iterable[Type[nn.Module]],
109 ignored_modules: Set[nn.Module],
[all …]
|
| D | _init_utils.py | 288 module: nn.Module,
289 ignored_modules: Optional[Iterable[torch.nn.Module]],
291 Optional[Iterable[torch.nn.Parameter]], Optional[Iterable[torch.nn.Module]]
364 module: nn.Module,
413 module: nn.Module,
552 def _verify_managed_params(module: nn.Module, params: List[nn.Parameter]) -> None:
574 fully_sharded_module: nn.Module,
576 param_init_fn: Optional[Callable[[nn.Module], None]],
644 fully_sharded_module: nn.Module,
672 root_module: nn.Module,
[all …]
|
| D | _wrap_utils.py | 26 root_module: nn.Module,
28 ignored_modules: Set[nn.Module],
104 def _check_nested_wrapping(root_module: nn.Module):
114 overridden_module_classes: Set[Type[nn.Module]],
127 root_module: nn.Module,
128 modules_to_wrap: Set[nn.Module],
180 root_module: nn.Module,
222 module_to_wrap: nn.Module,
224 visited_modules: Set[nn.Module],
|
| D | _state_dict_utils.py | 85 module: nn.Module, fsdp_state: _FSDPState
98 module: nn.Module, fsdp_state
110 module: nn.Module,
138 def _exit_unshard_params_ctx(module: nn.Module, fsdp_state: _FSDPState) -> None:
145 module: nn.Module,
158 module: nn.Module,
181 module: nn.Module,
288 module: nn.Module,
313 module: nn.Module,
358 module: nn.Module,
[all …]
|
| /external/starlark-go/starlarkstruct/ |
| D | module.go | 14 type Module struct { struct
15 Name string
16 Members starlark.StringDict
21 func (m *Module) Attr(name string) (starlark.Value, error) { return m.Members[name], nil }
22 func (m *Module) AttrNames() []string { return m.Members.Keys() }
23 func (m *Module) Freeze() { m.Members.Freeze() }
24 …ule) Hash() (uint32, error) { return 0, fmt.Errorf("unhashable: %s", m.Type()) }
25 …(m *Module) String() string { return fmt.Sprintf("<module %q>", m.Name) }
26 func (m *Module) Truth() starlark.Bool { return true }
27 func (m *Module) Type() string { return "module" }
|
| /external/executorch/examples/models/llama/source_transformation/ |
| D | pre_quantization.py | 23 module: torch.nn.Module,
29 def filter_fn(child: torch.nn.Module, cur_fqn: str) -> bool:
39 def replacement_fn(child: torch.nn.Module) -> torch.nn.Module:
61 module: torch.nn.Module,
87 module: torch.nn.Module,
91 def filter_fn(child: torch.nn.Module, cur_fqn: str) -> bool:
103 def replacement_fn(child: torch.nn.Module) -> torch.nn.Module:
121 module: torch.nn.Module,
138 module: torch.nn.Module,
144 def filter_fn(child: torch.nn.Module, cur_fqn: str) -> bool:
[all …]
|
| /external/pytorch/torch/ao/pruning/sparsifier/ |
| D | utils.py | 19 def module_contains_param(module: nn.Module, parametrization: Type[nn.Module]) -> bool:
30 mod: nn.Module, mapping: Dict[Type[nn.Module], Type[nn.Module]]
69 model: nn.Module, module: nn.Module, prefix: str = ""
83 def fqn_to_module(model: Optional[nn.Module], path: str) -> Optional[nn.Module]:
94 def get_arg_info_from_tensor_fqn(model: nn.Module, tensor_fqn: str) -> Dict[str, Any]:
|
| /external/bcc/src/cc/ |
| D | syms.h | 152 struct Module { struct
161 Module(const char *name, std::shared_ptr<ModulePath> path, argument
165 std::shared_ptr<ModulePath> path_; argument
169 ModuleType type_; argument
194 std::vector<Module> modules_; argument
222 struct Module { struct
223 Module(const char *module_name): in Module() function
239 std::unordered_map<std::string, std::unique_ptr<Module> > buildmap_; argument
|
| /external/bcc/src/cc/includes/ |
| D | syms.h | 152 struct Module { struct
161 Module(const char *name, std::shared_ptr<ModulePath> path, argument
165 std::shared_ptr<ModulePath> path_; argument
169 ModuleType type_; argument
194 std::vector<Module> modules_; argument
222 struct Module { struct
223 Module(const char *module_name): in Module() function
239 std::unordered_map<std::string, std::unique_ptr<Module> > buildmap_; argument
|
| /external/wayland-protocols/ |
| D | wayland_protocol_codegen.go | 184 type Module struct { struct
185 android.ModuleBase
186 android.DefaultableModuleBase
187 android.ApexModuleBase
189 android.ImageInterface
191 properties generatorProperties
193 taskGenerator taskFunc
195 rawCommands []string
197 exportedIncludeDirs android.Paths
199 outputFiles android.Paths
[all …]
|
| /external/pytorch/test/distributed/_composable/ |
| D | test_contract.py | 31 module: nn.Module, inp: Tuple[torch.Tensor]
36 module: nn.Module, inp: Tuple[torch.Tensor], out: torch.Tensor
41 module: nn.Module, grad_output: torch.Tensor
46 module: nn.Module,
53 def noop_api(module: nn.Module) -> nn.Module:
83 def wrap_module(module: nn.Module) -> nn.Module:
95 module: nn.Module, inp: Tuple[torch.Tensor]
104 def api(module: nn.Module) -> nn.Module:
119 def api1(module: nn.Module) -> nn.Module:
123 def api2(module: nn.Module) -> nn.Module:
[all …]
|
| /external/pytorch/torch/distributed/checkpoint/ |
| D | state_dict.py | 168 model: nn.Module,
236 def recurse(module: nn.Module, curr_fqn: str) -> Generator:
265 model: nn.Module,
269 submodules: Optional[Set[nn.Module]] = None,
439 def _state_dict_fn(obj: Union[nn.Module, torch.optim.Optimizer], api: str) -> Callable:
466 model: nn.Module, info: _StateDictInfo
534 model: nn.Module,
735 model: nn.Module,
798 model: nn.Module,
871 model: nn.Module,
[all …]
|
| /external/pytorch/torch/_functorch/ |
| D | make_functional.py | 72 mod: nn.Module,
97 mod: nn.Module,
111 mod: nn.Module,
117 mod: nn.Module,
134 mod: nn.Module, names_map: Dict[str, List[str]], elems: Iterable[Tensor]
148 mod: nn.Module,
158 model: nn.Module,
177 def make_functional_deprecated_v1(model: nn.Module):
219 def make_functional_with_buffers_deprecated_v1(model: nn.Module):
263 stateless_model: nn.Module,
[all …]
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/cxxbridge-cmd-1.0.131/src/syntax/ |
| D | file.rs | 10 pub(crate) struct Module { struct
12 pub cfg: CfgExpr,
13 pub namespace: Namespace,
14 pub attrs: Vec<Attribute>,
16 pub vis: Visibility,
17 pub unsafety: Option<Token![unsafe]>,
19 pub mod_token: Token![mod],
21 pub ident: Ident,
45 impl Parse for Module { implementation
|
| /external/rust/cxx/syntax/ |
| D | file.rs | 10 pub(crate) struct Module { struct
12 pub cfg: CfgExpr,
13 pub namespace: Namespace,
14 pub attrs: Vec<Attribute>,
16 pub vis: Visibility,
17 pub unsafety: Option<Token![unsafe]>,
19 pub mod_token: Token![mod],
21 pub ident: Ident,
45 impl Parse for Module { argument
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/cxxbridge-macro-1.0.130/src/syntax/ |
| D | file.rs | 10 pub(crate) struct Module { struct
12 pub cfg: CfgExpr,
13 pub namespace: Namespace,
14 pub attrs: Vec<Attribute>,
16 pub vis: Visibility,
17 pub unsafety: Option<Token![unsafe]>,
19 pub mod_token: Token![mod],
21 pub ident: Ident,
45 impl Parse for Module { implementation
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/cxxbridge-cmd-1.0.131/src/syntax/ |
| D | file.rs | 10 pub(crate) struct Module { struct
12 pub cfg: CfgExpr,
13 pub namespace: Namespace,
14 pub attrs: Vec<Attribute>,
16 pub vis: Visibility,
17 pub unsafety: Option<Token![unsafe]>,
19 pub mod_token: Token![mod],
21 pub ident: Ident,
45 impl Parse for Module { implementation
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/cxxbridge-macro-1.0.130/src/syntax/ |
| D | file.rs | 10 pub(crate) struct Module { struct
12 pub cfg: CfgExpr,
13 pub namespace: Namespace,
14 pub attrs: Vec<Attribute>,
16 pub vis: Visibility,
17 pub unsafety: Option<Token![unsafe]>,
19 pub mod_token: Token![mod],
21 pub ident: Ident,
45 impl Parse for Module { argument
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | prepare.py | 129 named_modules: Dict[str, torch.nn.Module],
235 node: Node, named_modules: Dict[str, torch.nn.Module]
367 named_modules: Dict[str, torch.nn.Module],
429 named_modules: Dict[str, torch.nn.Module],
462 root: torch.nn.Module, module_to_qat_module: Dict[Pattern, Type[torch.nn.Module]]
478 model: torch.nn.Module,
479 named_modules: Dict[str, torch.nn.Module],
511 named_modules: Dict[str, torch.nn.Module],
564 named_modules: Dict[str, torch.nn.Module],
665 named_modules: Dict[str, torch.nn.Module],
[all …]
|
| /external/pytorch/torch/ao/quantization/experimental/ |
| D | adaround_optimization.py | 19 model: Union[torch.nn.Module, torch.nn.DataParallel],
22 Union[torch.nn.Module, torch.nn.DataParallel],
24 Optional[torch.nn.Module],
37 feed_forward_wrapper: Optional[torch.nn.Module] = None,
97 self, module: torch.nn.Module, q_module: torch.nn.Module, data: List[Any]
150 q_module: torch.nn.Module,
169 module: torch.nn.Module,
170 q_module: torch.nn.Module,
|
| /external/rust/android-crates-io/crates/bindgen/ir/ |
| D | module.rs | 23 pub(crate) struct Module { struct
27 kind: ModuleKind, argument
32 impl Module { impl
63 impl DotAttributes for Module { implementation
76 impl ClangSubItemParser for Module { implementation
|
| /external/pytorch/torch/nn/utils/ |
| D | parametrize.py | 122 modules: Sequence[Module],
313 def _inject_new_class(module: Module) -> None:
362 def _inject_property(module: Module, tensor_name: str) -> None:
418 module: Module,
420 parametrization: Module,
631 def is_parametrized(module: Module, tensor_name: Optional[str] = None) -> bool:
654 module: Module,
746 def type_before_parametrizations(module: Module) -> type:
759 from_module: Module,
760 to_module: Module,
|
| /external/pytorch/torch/csrc/jit/api/ |
| D | module.h | 87 struct TORCH_API Module : public Object { struct
92 Module& operator=(const Module&) = default; argument
94 Module& operator=(Module&&) noexcept = default; argument
99 Module(ModulePtr module_value) : Object(std::move(module_value)) {} in Module() argument
104 "Module::set_optimized() is deprecated and has no effect. " in set_optimized() argument
110 "Module::is_optimized() is deprecated and always returns true. " in is_optimized() argument
123 void register_buffer(const std::string& name, at::Tensor v) { in register_buffer()
150 void register_module(const std::string& name, const Module& module) { in register_module() argument
187 void eval() { in eval()
191 bool is_training() const { in is_training()
[all …]
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | weight_utils.py | 22 def mod_weight_detach(mod: nn.Module) -> torch.Tensor:
26 def mod_0_weight_detach(mod: nn.Module) -> torch.Tensor:
30 def mod_weight_bias_0(mod: nn.Module) -> torch.Tensor:
34 def get_lstm_weight(mod: nn.Module) -> List[torch.Tensor]:
43 def get_qlstm_weight(mod: nn.Module) -> List[torch.Tensor]:
51 def get_conv_mod_weight(mod: nn.Module) -> torch.Tensor:
60 def get_linear_mod_weight(mod: nn.Module) -> torch.Tensor:
69 def get_lstm_mod_weights(mod: nn.Module) -> List[torch.Tensor]:
|