Home
last modified time | relevance | path

Searched refs:lazy_exponent (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp6449 torch::Tensor lazy_exponent = CopyToDevice(exponent, device); in TEST_F() local
6450 torch::Tensor lazy_result = torch::pow(lazy_base, lazy_exponent); in TEST_F()
6463 torch::Tensor lazy_exponent = CopyToDevice(exponent, device); in TEST_F() local
6464 torch::Tensor lazy_result = lazy_base.pow_(lazy_exponent); in TEST_F()
6478 torch::Tensor lazy_exponent = CopyToDevice(exponent, device); in TEST_F() local
6479 torch::Tensor lazy_result = torch::pow(lazy_base, lazy_exponent); in TEST_F()
6489 torch::Tensor lazy_exponent = CopyToDevice(exponent, device); in TEST_F() local
6490 torch::Tensor lazy_result = torch::pow(base, lazy_exponent); in TEST_F()