Searched defs:x_spt (Results 1 – 3 of 3) 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/dynamo/ |
| H A D | test_repros.py | 579 def forward(self, 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
|