Searched defs:lazy_d (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 3738 torch::Tensor lazy_d = torch::baddbmm(lazy_a, lazy_b, lazy_c, beta, alpha); in TEST_F() local
3757 torch::Tensor lazy_d = lazy_a.baddbmm_(lazy_b, lazy_c, beta, alpha); in TEST_F() local
3791 torch::Tensor lazy_d = CopyToDevice(d, device); in TEST_F() local
4125 torch::Tensor lazy_d = torch::addcmul(lazy_a, lazy_b, lazy_c, 3.1165); in TEST_F() local
4145 torch::Tensor lazy_d = torch::addcdiv(lazy_a, lazy_b, lazy_c, 3.1165); in TEST_F() local
4165 torch::Tensor lazy_d = torch::addcdiv(lazy_a, lazy_b, lazy_c, 3.1165); in TEST_F() local
4353 torch::Tensor lazy_d = torch::stack({lazy_a, lazy_b, lazy_c}, dim); in TEST_F() local
4372 torch::Tensor lazy_d = torch::cat({lazy_a, lazy_b, lazy_c}, dim); in TEST_F() local
4453 torch::Tensor lazy_d = torch::scatter(lazy_a, dim, lazy_c, lazy_b); in TEST_F() local
4473 torch::Tensor lazy_d = torch::scatter(lazy_a, 0, lazy_c, lazy_b); in TEST_F() local
[all …]
|