Searched refs:test_pow_scalar_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 3966 void test_pow_scalar_tensor(const float base, const at::IntArrayRef other) { in test_pow_scalar_tensor() function
3984 test_pow_scalar_tensor(2.5, {4}); // 1d in TEST_F()
3985 test_pow_scalar_tensor(2, {4, 2}); // 2d in TEST_F()
3986 test_pow_scalar_tensor(7.7, {11, 7, 9}); // 3d in TEST_F()
3987 test_pow_scalar_tensor(3, {3, 11, 9, 7}); // 4d in TEST_F()
|