Searched defs:lazy_index (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 5707 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5740 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5775 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5810 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5846 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5896 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5946 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
5995 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
6037 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
6089 torch::Tensor lazy_index = CopyToDevice(index, device); in TEST_F() local
[all …]
|