Searched refs:lazy_start (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 11461 torch::Tensor lazy_start = CopyToDevice(start, device); in TEST_F() local
11464 torch::Tensor lazy_res = torch::lerp(lazy_start, lazy_end, lazy_weight); in TEST_F()
11479 torch::Tensor lazy_start = CopyToDevice(start, device); in TEST_F() local
11481 torch::Tensor lazy_res = torch::lerp(lazy_start, lazy_end, weight); in TEST_F()
11538 torch::Tensor lazy_start = CopyToDevice(start, device); in TEST_F() local
11541 torch::Tensor lazy_res = torch::empty({3, 4}, lazy_start.options()); in TEST_F()
11542 torch::lerp_out(lazy_res, lazy_start, lazy_end, lazy_weight); in TEST_F()
11559 torch::Tensor lazy_start = CopyToDevice(start, device); in TEST_F() local
11561 torch::Tensor lazy_res = torch::empty({3, 4}, lazy_start.options()); in TEST_F()
11562 torch::lerp_out(lazy_res, lazy_start, lazy_end, weight); in TEST_F()
|