Searched defs:lazy_input (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 639 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
823 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
834 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
845 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
861 torch::Tensor lazy_input = CopyToDevice(input_copy, device); in TEST_F() local
872 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
888 torch::Tensor lazy_input = CopyToDevice(input_copy, device); in TEST_F() local
899 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
915 torch::Tensor lazy_input = CopyToDevice(input_copy, device); in TEST_F() local
926 torch::Tensor lazy_input = CopyToDevice(input, device); in TEST_F() local
[all …]
|