Searched defs:n_inner_iter (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/functorch/examples/maml_omniglot/ |
| H A D | maml-omniglot-transforms.py | 122 def loss_for_task(net, n_inner_iter, x_spt, y_spt, x_qry, y_qry): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 4186 def loss_for_task(net, n_inner_iter, use_transform, x_spt, y_spt, x_qry, y_qry): argument
|