Searched refs:tensor_pow_scalar (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | pow_test.cpp | 120 void tensor_pow_scalar(const Vals vals, const Pows pows, const torch::ScalarType valsDtype, const t… in tensor_pow_scalar() function
286 tensor_pow_scalar(ints, non_neg_ints, kInt, kInt); in TEST()
287 tensor_pow_scalar(ints, non_neg_longs, kInt, kInt); in TEST()
288 tensor_pow_scalar(ints, floats, kInt, kFloat); in TEST()
289 tensor_pow_scalar(ints, doubles, kInt, kDouble); in TEST()
293 tensor_pow_scalar(longs, non_neg_ints, kLong, kLong); in TEST()
294 tensor_pow_scalar(longs, non_neg_longs, kLong, kLong); in TEST()
295 tensor_pow_scalar(longs, floats, kLong, kFloat); in TEST()
296 tensor_pow_scalar(longs, doubles, kLong, kDouble); in TEST()
300 tensor_pow_scalar(floats, ints, kFloat, kDouble); in TEST()
[all …]
|