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 | 3917 void test_pow_tensor_scalar(const at::IntArrayRef input_shape, const float exp) { in test_pow_tensor_scalar() function
3935 test_pow_tensor_scalar({4}, 2.5); // 1d in TEST_F()
3936 test_pow_tensor_scalar({4, 2}, -1); // 2d in TEST_F()
3937 test_pow_tensor_scalar({11, 7, 9}, 7.7); // 3d in TEST_F()
3938 test_pow_tensor_scalar({3, 11, 9, 7}, -0.03); // 4d in TEST_F()
|