Searched refs:scalar_pow_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | pow_test.cpp | 164 void scalar_pow_tensor(const Vals vals, c10::ScalarType vals_dtype, const Pows pows, c10::ScalarTyp… in scalar_pow_tensor() function
314 scalar_pow_tensor(ints, c10::kInt, ints, c10::kInt); in TEST()
315 scalar_pow_tensor(ints, c10::kInt, longs, c10::kLong); in TEST()
316 scalar_pow_tensor(ints, c10::kInt, floats, c10::kFloat); in TEST()
317 scalar_pow_tensor(ints, c10::kInt, doubles, c10::kDouble); in TEST()
321 scalar_pow_tensor(longs, c10::kLong, longs, c10::kLong); in TEST()
322 scalar_pow_tensor(longs, c10::kLong, floats, c10::kFloat); in TEST()
323 scalar_pow_tensor(longs, c10::kLong, doubles, c10::kDouble); in TEST()
327 scalar_pow_tensor(floats, c10::kFloat, floats, c10::kFloat); in TEST()
328 scalar_pow_tensor(floats, c10::kFloat, doubles, c10::kDouble); in TEST()
[all …]
|