Home
last modified time | relevance | path

Searched defs:Module (Results 201 – 225 of 2267) sorted by relevance

12345678910>>...91

/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_unsqueeze.py32 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor, int]
50 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor, int]
71 module: torch.nn.Module,
H A Dtest_relu.py45 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
65 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
89 module: torch.nn.Module,
H A Dtest_cat.py53 self, module: torch.nn.Module, test_data: Tuple[tuple[torch.Tensor, ...], int]
73 self, module: torch.nn.Module, test_data: Tuple[tuple[torch.Tensor, ...], int]
95 module: torch.nn.Module,
H A Dtest_permute.py51 module: torch.nn.Module,
75 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
98 module: torch.nn.Module,
H A Dtest_mul.py67 self, module: torch.nn.Module, test_data: tuple[torch.Tensor, torch.Tensor]
88 self, module: torch.nn.Module, test_data: tuple[torch.Tensor, torch.Tensor]
112 module: torch.nn.Module,
H A Dtest_layer_norm.py71 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
92 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
114 module: torch.nn.Module,
H A Dtest_div.py99 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
118 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
140 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
H A Dtest_linear.py131 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
151 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
173 module: torch.nn.Module,
H A Dtest_avg_pool.py52 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
74 self, module: torch.nn.Module, test_data: Tuple[torch.tensor]
99 module: torch.nn.Module,
H A Dtest_sub.py40 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
60 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
82 module: torch.nn.Module,
H A Dtest_mm.py51 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
71 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
94 module: torch.nn.Module,
H A Dtest_add.py58 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
77 self, module: torch.nn.Module, test_data: Tuple[torch.Tensor]
98 module: torch.nn.Module,
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_trace_utils.py80 def __init__(self, root_module: nn.Module) -> None:
95 def patch_tracer(self, tracer: torch.fx.Tracer, root_module: nn.Module):
120 module: nn.Module,
/aosp_15_r20/build/blueprint/
H A Dtransition.go109 Module() Module methodSpec
139 Module() Module methodSpec
267 func (c *outgoingTransitionContextImpl) Module() Module { func
279 func (c *incomingTransitionContextImpl) Module() Module { func
/aosp_15_r20/frameworks/base/services/core/jni/BroadcastRadio/
H A DBroadcastRadioService.cpp72 struct Module { struct
73 sp<V1_0::IBroadcastRadio> radioModule;
74 HalRevision halRev;
75 std::vector<hardware::broadcastradio::V1_0::BandConfig> bands;
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py884 def aot_module(mod: nn.Module, *args, **kwargs) -> nn.Module:
937 mod: nn.Module,
1111 mod: nn.Module,
1491 def _detect_attribute_assignment(mod: torch.nn.Module):
/aosp_15_r20/external/starlark-go/resolve/
H A Dbinding.go56 type Module struct { struct
57 Locals []*Binding // the file's (comprehension-)local variables
58 Globals []*Binding // the file's global variables
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_graph_partition.py24 self, module: torch.nn.Module, inputs: Tuple[torch.Tensor]
43 supported_modules: List[torch.nn.Module],
70 supported_modules: List[torch.nn.Module],
/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_param_group.py105 modules: Tuple[nn.Module, ...],
310 self, module: nn.Module, args: Tuple[Any, ...], kwargs: Dict[str, Any]
321 def post_forward(self, module: nn.Module, input: Any, output: Any):
590 params: List[nn.Parameter], modules: Tuple[nn.Module, ...]
/aosp_15_r20/external/executorch/extension/export_util/
H A Dutils.py27 model: Union[torch.fx.GraphModule, torch.nn.Module],
76 model: Union[torch.fx.GraphModule, torch.nn.Module],
100 model: Union[torch.fx.GraphModule, torch.nn.Module],
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dstage.py89 submodule: torch.nn.Module,
781 stage_module: torch.nn.Module,
1044 stage_module: torch.nn.Module,
1159 stage_modules: List[nn.Module],
1251 submodule: nn.Module,
/aosp_15_r20/prebuilts/build-tools/common/py3-stdlib/unittest/test/
H A Dtest_discovery.py159 class Module(object): class
233 class Module(object): class
318 class Module(object): class
435 class Module(object): class
720 class Module(object): class
/aosp_15_r20/external/python/cpython3/Lib/unittest/test/
Dtest_discovery.py159 class Module(object): class
233 class Module(object): class
318 class Module(object): class
435 class Module(object): class
720 class Module(object): class
/aosp_15_r20/packages/modules/Bluetooth/system/gd/
Dmodule.h36 class Module; variable
53 friend Module; variable
124 friend Module; variable
/aosp_15_r20/external/executorch/backends/qualcomm/utils/
H A Dutils.py173 def convert_linear_to_conv2d(module: torch.nn.Module):
198 def replace_linear(module: torch.nn.Module):
337 module: torch.nn.Module,
434 nn_module: torch.nn.Module,

12345678910>>...91