Searched refs:convert_float_half (Results 1 – 7 of 7) sorted by relevance
96 return convert_float_half(result0, result1); in lerp_scalar_kernel()143 return convert_float_half(result0, result1); in lerp_tensor_kernel()
25 auto res = at::vec::convert_float_half(src, src); in _store()79 Vectorized<Half> val = convert_float_half(val0, val1); in store()
79 result[0] = convert_float_half(src[0], src[1]);
144 inline Vectorized<Half> convert_float_half( in convert_float_half() function
1118 inline Vectorized<Half> convert_float_half(const Vectorized<float>& a, const Vectorized<float>& b) {
45 return convert_float_half(a, b);