Searched defs:ParamsT (Results 1 – 15 of 15) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | optimizer.py | 311 ParamsT: TypeAlias = Union[Iterable[torch.Tensor], Iterable[Dict[str, Any]]] variable
343 def __init__(self, params: ParamsT, defaults: Dict[str, Any]) -> None: # noqa: D107
|
| H A D | sparse_adam.py | 17 params: ParamsT,
|
| H A D | lbfgs.py | 219 params: ParamsT,
|
| H A D | rprop.py | 31 params: ParamsT,
|
| H A D | adadelta.py | 30 params: ParamsT,
|
| H A D | adamax.py | 31 params: ParamsT,
|
| H A D | asgd.py | 31 params: ParamsT,
|
| H A D | radam.py | 32 params: ParamsT,
|
| H A D | rmsprop.py | 31 params: ParamsT,
|
| H A D | _adafactor.py | 24 params: ParamsT,
|
| H A D | nadam.py | 33 params: ParamsT,
|
| H A D | adagrad.py | 28 params: ParamsT,
|
| H A D | adamw.py | 35 params: ParamsT,
|
| H A D | adam.py | 35 params: ParamsT,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/ |
| H A D | TunableOp.h | 33 virtual TuningStatus Call(const ParamsT*) { in Call()
|