Home
last modified time | relevance | path

Searched defs:AdamW (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/optim/_multi_tensor/
H A D__init__.py21 AdamW = partialclass(optim.AdamW, foreach=True) variable
/aosp_15_r20/external/pytorch/torch/optim/
H A D__init__.py15 from torch.optim.adamw import AdamW as AdamW unknown
H A Dadamw.py32 class AdamW(Optimizer): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Doptim_baseline.h827 inline std::vector<std::vector<torch::Tensor>> AdamW() { in AdamW() function