| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/examples/ |
| H A D | stateful_example.py | 42 def _make_stateful(model, optim): argument
47 def _train(model, optim, train_steps=1): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_optim.py | 346 def closure(optim, w, b, i): argument
1359 def fwd_bwd(optim, w, b, i): argument
1435 def fwd_bwd(optim, mod, i): argument
1480 def fwd_bwd(optim, w, b, i): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/ |
| H A D | test_tp_examples.py | 57 optim: Optional[Dict] = None variable in ExpCommCounts
252 def _validate_optim_step( argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/schedulers/ |
| H A D | step_lr.cpp | 4 namespace optim { namespace
|
| H A D | lr_scheduler.cpp | 5 namespace optim { namespace
|
| H A D | reduce_on_plateau_scheduler.cpp | 6 namespace optim { namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/optim/ |
| H A D | utils.py | 32 def register_functional_optim(key, optim): argument
|
| /aosp_15_r20/external/libcups/config-scripts/ |
| H A D | cups-compiler.m4 | 15 AC_ARG_WITH(optim, [ --with-optim set optimization flags ], optwith
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_optim/ |
| H A D | test_sharded_optim.py | 6 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | serialize.cpp | 13 namespace optim { namespace
|
| H A D | sgd.cpp | 15 namespace optim { namespace
|
| H A D | rmsprop.cpp | 13 namespace optim { namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_optim/ |
| H A D | api.py | 4 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/functorch/examples/maml_omniglot/ |
| H A D | maml-omniglot-higher.py | 42 import torch.optim as optim namespace
|
| H A D | maml-omniglot-ptonly.py | 41 import torch.optim as optim namespace
|
| H A D | maml-omniglot-transforms.py | 42 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/functorch/examples/dp_cifar10/ |
| H A D | cifar10_opacus.py | 23 import torch.optim as optim namespace
|
| H A D | cifar10_transforms.py | 22 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ |
| H A D | test_mnist.py | 13 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/functorch/examples/compilation/ |
| H A D | linear_train.py | 60 optim = torch.optim.SGD( variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/consistency_integration_test/ |
| H A D | keras_integration_tests.py | 152 def train_one_step(a, x, y, optim): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_memory.py | 7 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/rl/ |
| H A D | agent.py | 10 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | numeric_utils.py | 11 import torch.optim as optim namespace
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_optimizers.py | 51 def _assert_optimizer( argument
|