Searched defs:lazy_params (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 4837 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
4868 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
4970 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5010 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5050 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5088 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5126 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5158 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5202 torch::Tensor lazy_params = CopyToDevice(params, device); in TEST_F() local
5242 torch::Tensor lazy_params = CopyToDevice(params.clone(), device); in TEST_F() local
[all …]
|