Searched defs:enable_nested_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 322 …def test_transformerencoder_fastpath(self, device, use_torchscript, enable_nested_tensor, use_auto… argument
413 …def test_transformerencoder_square_input(self, with_no_grad, training, enable_nested_tensor, devic… argument
458 def test_transformerencoder(self, batch_first, training, enable_nested_tensor, device): argument
488 def _test(batch_first, training, enable_nested_tensor): argument
|