| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | adamax.py | 40 capturable: bool = False,
63 capturable=capturable,
150 capturable = group["capturable"]
170 capturable=capturable,
240 capturable: bool,
251 if not torch._utils.is_compiling() and capturable:
286 if capturable:
314 capturable: bool,
323 if not torch._utils.is_compiling() and capturable:
394 if capturable:
[all …]
|
| H A D | rprop.py | 36 capturable: bool = False,
55 capturable=capturable,
160 capturable=group["capturable"],
234 capturable: bool,
246 if not torch._utils.is_compiling() and capturable:
265 if capturable:
280 if capturable:
302 capturable: bool,
312 if not torch._utils.is_compiling() and capturable:
366 if capturable:
[all …]
|
| H A D | adadelta.py | 37 capturable: bool = False,
58 capturable=capturable,
155 capturable,
184 capturable=capturable,
258 capturable: bool,
262 if not torch._utils.is_compiling() and capturable:
312 capturable: bool,
318 if not torch._utils.is_compiling() and capturable:
392 if capturable and isinstance(lr, torch.Tensor):
408 capturable: bool = False,
[all …]
|
| H A D | asgd.py | 40 capturable: bool = False,
58 capturable=capturable,
164 capturable=group["capturable"],
210 capturable: bool,
222 if not torch._utils.is_compiling() and capturable:
246 if capturable:
255 if capturable or mu.item() != 1:
260 if capturable:
286 capturable: bool,
295 if not torch._utils.is_compiling() and capturable:
[all …]
|
| H A D | adamw.py | 44 capturable: bool = False,
49 if foreach and not capturable:
73 capturable=capturable,
235 capturable=group["capturable"],
334 capturable: bool,
353 if not torch._utils.is_compiling() and capturable:
378 if capturable or differentiable:
453 capturable: bool,
460 if isinstance(lr, Tensor) and not capturable:
466 if not torch._utils.is_compiling() and capturable:
[all …]
|
| H A D | adam.py | 44 capturable: bool = False,
49 if foreach and not capturable:
74 capturable=capturable,
239 capturable=group["capturable"],
338 capturable: bool,
356 if not torch._utils.is_compiling() and capturable:
381 if capturable or differentiable:
457 capturable: bool,
463 if isinstance(lr, Tensor) and not capturable:
469 if not torch._utils.is_compiling() and capturable:
[all …]
|
| H A D | radam.py | 41 capturable: bool = False,
64 capturable=capturable,
165 capturable=group["capturable"],
269 capturable: bool,
279 if not torch._utils.is_compiling() and capturable:
294 step = step_t if capturable else _get_value(step_t)
335 if capturable:
368 capturable: bool,
377 if not torch._utils.is_compiling() and capturable:
428 if capturable:
[all …]
|
| H A D | nadam.py | 43 capturable: bool = False,
69 capturable=capturable,
204 capturable=group["capturable"],
295 capturable: bool,
313 if not torch._utils.is_compiling() and capturable:
326 if capturable:
352 if differentiable or capturable:
389 capturable: bool,
399 if not torch._utils.is_compiling() and capturable:
475 if capturable:
[all …]
|
| H A D | rmsprop.py | 38 capturable=False, argument
63 capturable=capturable,
191 capturable=group["capturable"],
279 capturable: bool,
286 if not torch._utils.is_compiling() and capturable:
350 capturable: bool,
359 if not torch._utils.is_compiling() and capturable:
445 if capturable and isinstance(lr, torch.Tensor):
455 if capturable and isinstance(lr, torch.Tensor):
475 capturable: bool = False,
[all …]
|
| H A D | optimizer.py | 739 capturable = False
744 capturable = pg["capturable"] if "capturable" in pg else False
747 if capturable or fused:
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | training_loss.py | 157 optimizer = optimizer_cls(model.parameters(), lr=args.lr, capturable=True)
|
| H A D | common.py | 2132 params, lr=0.01, capturable=True, foreach=True
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | complex_numbers.rst | 153 and capturable vs default optimizers. For more details, see https://pytorch.org/docs/stable/notes/n…
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_padding.py | 40 return torch.optim.Adam(m.parameters(), lr=0.01, capturable=True, foreach=True)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_optim.py | 1601 capturable = state_dict_cpu["param_groups"][0].get("capturable", False)
1610 "cuda" if capturable or fused else "cpu",
|
| H A D | test_cuda.py | 2903 opt = optimizer_ctor(params_control, capturable=False, **kwargs)
2912 opt = optimizer_ctor(params_graphed, capturable=True, **kwargs)
|
| /aosp_15_r20/external/pytorch/docs/source/notes/ |
| H A D | cuda.rst | 965 A set of ops is *capturable* if it doesn't violate any of the following constraints.
1007 If your entire network is capturable, you can capture and replay an entire iteration::
1081 network isn't capturable end-to-end, but
|