Home
last modified time | relevance | path

Searched defs:lazy_b (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp101 torch::Tensor lazy_b = lazy_a.clone(); in TEST_F() local
147 torch::Tensor lazy_b = torch::_cast_Byte(lazy_a); in TEST_F() local
160 torch::Tensor lazy_b = torch::_cast_Char(lazy_a); in TEST_F() local
173 torch::Tensor lazy_b = torch::_cast_Short(lazy_a); in TEST_F() local
186 torch::Tensor lazy_b = torch::_cast_Int(lazy_a); in TEST_F() local
199 torch::Tensor lazy_b = torch::_cast_Long(lazy_a); in TEST_F() local
212 torch::Tensor lazy_b = torch::_cast_Float(lazy_a); in TEST_F() local
220 torch::Tensor lazy_b = torch::ones( in TEST_F() local
249 torch::Tensor lazy_b = CopyToDevice(b, device); in TEST_F() local
263 torch::Tensor lazy_b = CopyToDevice(b, device); in TEST_F() local
[all …]