Home
last modified time | relevance | path

Searched defs:Adam (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/optim/_multi_tensor/
D__init__.py20 Adam = partialclass(optim.Adam, foreach=True) variable
/external/pytorch/torch/optim/
D__init__.py13 from torch.optim.adam import Adam as Adam unknown
Dadam.py32 class Adam(Optimizer): class
/external/pytorch/test/cpp/api/
Doptim_baseline.h349 inline std::vector<std::vector<torch::Tensor>> Adam() { in Adam() function
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadam.py32 class Adam(optimizer_v2.OptimizerV2): class
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py468 class Adam(Optimizer): class
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_utils.py678 class Adam(_Optimizer): class
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_test.cc240 static void Adam(int32_t n, Graph** init_g, Graph** train_g) { in Adam() function