Searched refs: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
3739 AllClose(d, lazy_d, /*rtol=*/1e-3, /*atol=*/1e-4); in TEST_F()
3757 torch::Tensor lazy_d = lazy_a.baddbmm_(lazy_b, lazy_c, beta, alpha); in TEST_F() local
3758 AllClose(d, lazy_d, /*rtol=*/1e-3, /*atol=*/1e-4); in TEST_F()
3791 torch::Tensor lazy_d = CopyToDevice(d, device); in TEST_F() local
3793 torch::chain_matmul({lazy_a, lazy_b, lazy_c, lazy_d}); in TEST_F()
4125 torch::Tensor lazy_d = torch::addcmul(lazy_a, lazy_b, lazy_c, 3.1165); in TEST_F() local
4126 AllClose(d, lazy_d); in TEST_F()
4145 torch::Tensor lazy_d = torch::addcdiv(lazy_a, lazy_b, lazy_c, 3.1165); in TEST_F() local
4146 AllClose(d, lazy_d); in TEST_F()
[all …]
|