Searched refs:test_pow_tensor_scalar_ (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 3941 void test_pow_tensor_scalar_(const at::IntArrayRef input_shape, const float exp) { in test_pow_tensor_scalar_() function
3960 test_pow_tensor_scalar_({4}, 2.5); // 1d in TEST_F()
3961 test_pow_tensor_scalar_({4, 2}, -1); // 2d in TEST_F()
3962 test_pow_tensor_scalar_({11, 7, 9}, 7.7); // 3d in TEST_F()
3963 test_pow_tensor_scalar_({3, 11, 9, 7}, -0.03); // 4d in TEST_F()
|