Searched refs:convert_half_float (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | LerpKernel.cpp | 92 auto [self_vec0, self_vec1] = convert_half_float(self_vec); in lerp_scalar_kernel()
93 auto [end_vec0, end_vec1] = convert_half_float(end_vec); in lerp_scalar_kernel()
138 auto [self_vec0, self_vec1] = convert_half_float(self_vec); in lerp_tensor_kernel()
139 auto [end_vec0, end_vec1] = convert_half_float(end_vec); in lerp_tensor_kernel()
140 auto [weight_vec0, weight_vec1] = convert_half_float(weight_vec); in lerp_tensor_kernel()
|
| D | utils.h | 68 auto [v0, v1] = convert_half_float(Vectorized<Half>::loadu(ptr)); in loadu()
105 return convert_half_float(Vectorized<Half>::loadu(ptr));
118 return convert_half_float(Vectorized<Half>::loadu(ptr, count));
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec512/ |
| D | vec512_convert.h | 88 std::tie(result[0], result[1]) = convert_half_float(src[0]);
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec256/vsx/ |
| D | vec256_common_vsx.h | 132 inline std::tuple<Vectorized<float>, Vectorized<float>> convert_half_float( in convert_half_float() function
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec256/ |
| D | vec256_convert.h | 88 std::tie(result[0], result[1]) = convert_half_float(src[0]);
|
| D | vec256_bfloat16.h | 1101 inline std::tuple<Vectorized<float>, Vectorized<float>> convert_half_float(const Vectorized<Half>& …
|
| /external/mesa3d/src/mesa/main/ |
| D | format_utils.c | 913 convert_half_float(void *void_dst, int num_dst_channels, in convert_half_float() function
1432 convert_half_float(void_dst, num_dst_channels, void_src, src_type, in _mesa_swizzle_and_convert()
|
| /external/pytorch/aten/src/ATen/cpu/vec/ |
| D | functional_bfloat16.h | 31 return convert_half_float(a);
|