Home
last modified time | relevance | path

Searched defs:Optimizer (Results 1 – 25 of 185) sorted by relevance

12345678

/aosp_15_r20/external/mesa3d/src/asahi/compiler/test/
H A Dtest-optimizer.cpp54 class Optimizer : public testing::Test { class
56 Optimizer() in Optimizer() function in Optimizer
79 TEST_F(Optimizer, FloatCopyprop) in TEST_F() argument
88 TEST_F(Optimizer, FloatConversion) in TEST_F() argument
107 TEST_F(Optimizer, FusedFABSNEG) in TEST_F() argument
116 TEST_F(Optimizer, FusedFabsAbsorb) in TEST_F() argument
122 TEST_F(Optimizer, FusedFnegCancel) in TEST_F() argument
131 TEST_F(Optimizer, FusedNot) in TEST_F() argument
172 TEST_F(Optimizer, FmulFsatF2F16) in TEST_F() argument
183 TEST_F(Optimizer, Copyprop) in TEST_F() argument
[all …]
/aosp_15_r20/external/mesa3d/src/panfrost/compiler/test/
H A Dtest-optimizer.cpp57 class Optimizer : public testing::Test { class
59 Optimizer() in Optimizer() function in Optimizer
80 TEST_F(Optimizer, FusedFABSNEG) in TEST_F() argument
98 TEST_F(Optimizer, FusedFABSNEGForFP16) in TEST_F() argument
107 TEST_F(Optimizer, FuseFADD_F32WithEqualSourcesAbsAbsAndClamp) in TEST_F() argument
143 TEST_F(Optimizer, FuseFADD_V2F16WithDifferentSourcesAbsAbsAndClamp) in TEST_F() argument
179 TEST_F(Optimizer, AvoidFADD_V2F16WithEqualSourcesAbsAbsAndClamp) in TEST_F() argument
200 TEST_F(Optimizer, SwizzlesComposedForFP16) in TEST_F() argument
230 TEST_F(Optimizer, PreserveWidens) in TEST_F() argument
244 TEST_F(Optimizer, DoNotMixSizesForFABSNEG) in TEST_F() argument
[all …]
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dstate_dict.py266 optims: Tuple[torch.optim.Optimizer, ...],
439 def _state_dict_fn(obj: Union[nn.Module, torch.optim.Optimizer], api: str) -> Callable:
586 def _init_optim_state(optim: torch.optim.Optimizer) -> None:
689 optim: torch.optim.Optimizer,
736 optimizers: Tuple[torch.optim.Optimizer, ...],
799 optim: torch.optim.Optimizer,
872 optimizers: Tuple[torch.optim.Optimizer, ...],
1002 optimizers: Union[torch.optim.Optimizer, Iterable[torch.optim.Optimizer]],
1047 optimizers: Union[torch.optim.Optimizer, Iterable[torch.optim.Optimizer]],
1208 optimizers: Union[torch.optim.Optimizer, Iterable[torch.optim.Optimizer]],
[all …]
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/test/opt/
H A Doptimizer_test.cpp38 TEST(Optimizer, CanRunNullPassWithDistinctInputOutputVectors) { in TEST() argument
55 TEST(Optimizer, CanRunTransformingPassWithDistinctInputOutputVectors) { in TEST() argument
71 TEST(Optimizer, CanRunNullPassWithAliasedVectors) { in TEST() argument
85 TEST(Optimizer, CanRunNullPassWithAliasedVectorDataButDifferentSize) { in TEST() argument
107 TEST(Optimizer, CanRunTransformingPassWithAliasedVectors) { in TEST() argument
121 TEST(Optimizer, CanValidateFlags) { in TEST() argument
130 TEST(Optimizer, CanRegisterPassesFromFlags) { in TEST() argument
225 TEST(Optimizer, RemoveNop) { in TEST() argument
273 TEST(Optimizer, AvoidIntegrityCheckForExtraLineInfo) { in TEST() argument
332 TEST(Optimizer, AvoidIntegrityCheckForDebugScope) { in TEST() argument
/aosp_15_r20/external/angle/third_party/spirv-tools/src/test/opt/
H A Doptimizer_test.cpp38 TEST(Optimizer, CanRunNullPassWithDistinctInputOutputVectors) { in TEST() argument
55 TEST(Optimizer, CanRunTransformingPassWithDistinctInputOutputVectors) { in TEST() argument
71 TEST(Optimizer, CanRunNullPassWithAliasedVectors) { in TEST() argument
85 TEST(Optimizer, CanRunNullPassWithAliasedVectorDataButDifferentSize) { in TEST() argument
107 TEST(Optimizer, CanRunTransformingPassWithAliasedVectors) { in TEST() argument
121 TEST(Optimizer, CanValidateFlags) { in TEST() argument
130 TEST(Optimizer, CanRegisterPassesFromFlags) { in TEST() argument
225 TEST(Optimizer, RemoveNop) { in TEST() argument
273 TEST(Optimizer, AvoidIntegrityCheckForExtraLineInfo) { in TEST() argument
332 TEST(Optimizer, AvoidIntegrityCheckForDebugScope) { in TEST() argument
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/test/opt/
Doptimizer_test.cpp38 TEST(Optimizer, CanRunNullPassWithDistinctInputOutputVectors) { in TEST() argument
55 TEST(Optimizer, CanRunTransformingPassWithDistinctInputOutputVectors) { in TEST() argument
71 TEST(Optimizer, CanRunNullPassWithAliasedVectors) { in TEST() argument
85 TEST(Optimizer, CanRunNullPassWithAliasedVectorDataButDifferentSize) { in TEST() argument
107 TEST(Optimizer, CanRunTransformingPassWithAliasedVectors) { in TEST() argument
121 TEST(Optimizer, CanValidateFlags) { in TEST() argument
130 TEST(Optimizer, CanRegisterPassesFromFlags) { in TEST() argument
225 TEST(Optimizer, RemoveNop) { in TEST() argument
273 TEST(Optimizer, AvoidIntegrityCheckForExtraLineInfo) { in TEST() argument
332 TEST(Optimizer, AvoidIntegrityCheckForDebugScope) { in TEST() argument
/aosp_15_r20/external/pytorch/torch/optim/
H A Dlr_scheduler.py71 def _format_param(name: str, optimizer: Optimizer, param):
95 self, optimizer: Optimizer, last_epoch=-1, verbose="deprecated"
124 def patch_track_step_called(opt: Optimizer):
320 optimizer: Optimizer,
420 optimizer: Optimizer,
524 optimizer: Optimizer,
583 optimizer: Optimizer,
650 optimizer: Optimizer,
728 optimizer: Optimizer,
806 self, optimizer: Optimizer, gamma: float, last_epoch=-1, verbose="deprecated"
[all …]
H A Doptimizer.py318 class Optimizer: class
572 self, hook: Callable[["Optimizer"], None], prepend: bool = False
605 hook: Callable[["Optimizer", StateDict], Optional[StateDict]],
759 hook: Callable[["Optimizer", StateDict], Optional[StateDict]],
797 self, hook: Callable[["Optimizer"], None], prepend: bool = False
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dfully_sharded_data_parallel.py1260 optim: torch.optim.Optimizer,
1324 optim: Optional[torch.optim.Optimizer] = None,
1377 optim: torch.optim.Optimizer,
1453 optim: torch.optim.Optimizer,
1494 optim: Optional[torch.optim.Optimizer] = None,
1567 optim: torch.optim.Optimizer,
1612 optim: Optional[torch.optim.Optimizer] = None,
1697 optim: Optional[torch.optim.Optimizer] = None,
1814 optim: torch.optim.Optimizer,
1910 optim: torch.optim.Optimizer,
H A D_optim_utils.py420 optim: Optional[torch.optim.Optimizer] = None,
880 optim: torch.optim.Optimizer,
1070 optim: torch.optim.Optimizer,
1121 optim: torch.optim.Optimizer,
1871 optim: torch.optim.Optimizer,
H A Dsharded_grad_scaler.py203 optimizer: torch.optim.Optimizer,
264 def unscale_(self, optimizer: torch.optim.Optimizer) -> None:
/aosp_15_r20/external/pytorch/test/
H A Dtest_optim.py1620 def _state_dict_pre_hook(optimizer: Optimizer) -> None:
1625 optimizer: Optimizer, state_dict: Dict[str, Any]
1677 optimizer: Optimizer, state_dict: Dict[str, Any]
1683 optimizer: Optimizer, state_dict: Dict[str, Any]
1692 def _load_state_dict_post_hook(optimizer: Optimizer) -> None:
1762 def post_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1794 def pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1826 def global_pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1830 def global_post_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
1834 def local_pre_hook(opt: Optimizer, args: Tuple[Any], kwargs: Dict[Any, Any]):
[all …]
/aosp_15_r20/external/pytorch/torch/amp/
H A Dgrad_scaler.py236 optimizer: torch.optim.Optimizer,
287 def unscale_(self, optimizer: torch.optim.Optimizer) -> None:
345 optimizer: torch.optim.Optimizer,
356 self, optimizer: torch.optim.Optimizer, *args: Any, **kwargs: Any
672 def _check_inf_per_device(self, optimizer: torch.optim.Optimizer) -> Dict[str, Any]:
684 def _found_inf_per_device(self, optimizer: torch.optim.Optimizer) -> Dict[str, Any]:
/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_state_dict.py162 optimizer_class: Type[Optimizer],
241 optimizer_class: Type[Optimizer],
281 def _test_ddp(self, use_composable: bool, optimizer_class: Type[Optimizer]) -> None:
309 optimizer_class: Type[Optimizer],
392 def _test_single_gpu(self, optimizer_class: Type[Optimizer]) -> None:
434 self, optimizer_class: Type[Optimizer]
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dcommon_state_dict.py58 optim: torch.optim.Optimizer,
111 optim: torch.optim.Optimizer,
112 new_optim: torch.optim.Optimizer,
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcompile.py44 optimizer: Optional[torch.optim.Optimizer] = None,
80 optimizer: Optional[torch.optim.Optimizer] = None,
123 optimizer: Optional[torch.optim.Optimizer] = None,
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_use_orig_params.py90 optim_class: Type[torch.optim.Optimizer],
120 optim_class: Type[torch.optim.Optimizer],
167 ddp_optim: torch.optim.Optimizer,
169 fsdp_optim: torch.optim.Optimizer,
338 optim_class: Type[torch.optim.Optimizer],
H A Dtest_fsdp_optim_state.py324 optim_class: Type[torch.optim.Optimizer] = torch.optim.Adam,
358 optim_class: Type[torch.optim.Optimizer] = torch.optim.Adam,
382 optim: torch.optim.Optimizer,
/aosp_15_r20/external/pytorch/torch/distributed/_tools/
H A Dfsdp2_mem_tracker.py158 optm: Optional[torch.optim.Optimizer] = None,
406 optimizer: optim.Optimizer, args: Any, kwargs: Any
411 optimizer: optim.Optimizer, args: Any, kwargs: Any
570 self, *external: Union[nn.Module, optim.Optimizer, torch.Tensor]
H A Dmem_tracker.py770 self, reftype: _RefType, optimizer: optim.Optimizer
785 optimizer: optim.Optimizer, args: Any, kwargs: Any
790 optimizer: optim.Optimizer, args: Any, kwargs: Any
815 self, *external: Union[nn.Module, optim.Optimizer, torch.Tensor]
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_clip_grad_norm_.py29 ref_optim: torch.optim.Optimizer,
31 optim: torch.optim.Optimizer,
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/
H A Doptimizer.h117 explicit Optimizer( in Optimizer() function
127 explicit Optimizer( in Optimizer() function
/aosp_15_r20/external/pytorch/test/distributed/_tools/
H A Dtest_mem_tracker.py157 model: nn.Module, opt: torch.optim.Optimizer
186 opt: torch.optim.Optimizer,
/aosp_15_r20/frameworks/base/tools/aapt2/cmd/
H A DOptimize.cpp129 class Optimizer { class
131 Optimizer(OptimizeContext* context, const OptimizeOptions& options) in Optimizer() function in aapt::Optimizer
/aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/
H A Dtest_2d_composability.py254 optimizer_class: Type[torch.optim.Optimizer],
258 _model: nn.Module, _optim: torch.optim.Optimizer, _inp: torch.Tensor

12345678