Searched refs:convert_bfloat16_float (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | LerpKernel.cpp | 75 auto [self_vec0, self_vec1] = convert_bfloat16_float(self_vec); in lerp_scalar_kernel()
76 auto [end_vec0, end_vec1] = convert_bfloat16_float(end_vec); in lerp_scalar_kernel()
123 auto [self_vec0, self_vec1] = convert_bfloat16_float(self_vec); in lerp_tensor_kernel()
124 auto [end_vec0, end_vec1] = convert_bfloat16_float(end_vec); in lerp_tensor_kernel()
125 auto [weight_vec0, weight_vec1] = convert_bfloat16_float(weight_vec); in lerp_tensor_kernel()
|
| D | WeightNormKernel.cpp | 73 auto [v_fvec0, v_fvec1] = convert_bfloat16_float(v_bvec); in sum_norm_per_row()
111 auto [v_fvec0, v_fvec1] = convert_bfloat16_float(v_bvec); in apply_norm_per_row()
250 auto [grad_w_fvec0, grad_w_fvec1] = convert_bfloat16_float(grad_w_bvec); in sum_product_per_row()
252 auto [v_fvec0, v_fvec1] = convert_bfloat16_float(v_bvec); in sum_product_per_row()
297 auto [grad_w_fvec0, grad_w_fvec1] = convert_bfloat16_float(grad_w_bvec); in apply_per_row_backward()
299 auto [v_fvec0, v_fvec1] = convert_bfloat16_float(v_bvec); in apply_per_row_backward()
|
| D | utils.h | 64 auto [v0, v1] = convert_bfloat16_float(Vectorized<BFloat16>::loadu(ptr)); in loadu()
101 return convert_bfloat16_float(Vectorized<BFloat16>::loadu(ptr));
114 return convert_bfloat16_float(Vectorized<BFloat16>::loadu(ptr, count));
|
| D | PointwiseOpsKernel.cpp | 120 auto [input0, input1] = convert_bfloat16_float(input); in smooth_l1_backward_cpu_kernel()
121 auto [target0, target1] = convert_bfloat16_float(target); in smooth_l1_backward_cpu_kernel()
122 auto [grad_output0, grad_output1] = convert_bfloat16_float(grad_output); in smooth_l1_backward_cpu_kernel()
|
| D | BinaryOpsKernel.cpp | 348 auto [a0, a1] = convert_bfloat16_float(a); in remainder_kernel()
349 auto [b0, b1] = convert_bfloat16_float(b); in remainder_kernel()
748 auto [a0, a1] = convert_bfloat16_float(a); in smooth_l1_kernel()
749 auto [b0, b1] = convert_bfloat16_float(b); in smooth_l1_kernel()
834 auto [a0, a1] = convert_bfloat16_float(a); in sigmoid_backward_kernel()
835 auto [b0, b1] = convert_bfloat16_float(b); in sigmoid_backward_kernel()
|
| D | ReduceOpsKernel.cpp | 185 auto [data_fvec0, data_fvec1] = convert_bfloat16_float(data_bvec); in norm_two_reduce_step()
|
| D | AdaptiveMaxPoolKernel.cpp | 768 auto [val_fvec0, val_fvec1] = convert_bfloat16_float(val_bvec); in cpu_adaptive_max_pool3d_channels_last()
|
| D | AvgPoolKernel.cpp | 871 auto [data_fvec0, data_fvec1] = convert_bfloat16_float(data_bvec); in cpu_avg_pool3d_channels_last()
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec256/vsx/ |
| D | vec256_bfloat16_vsx.h | 13 inline std::tuple<Vectorized<float>, Vectorized<float>> convert_bfloat16_float( in convert_bfloat16_float() function
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec512/ |
| D | vec512_convert.h | 61 std::tie(result[0], result[1]) = convert_bfloat16_float(src[0]);
|
| /external/pytorch/aten/src/ATen/cpu/vec/vec256/ |
| D | vec256_convert.h | 61 std::tie(result[0], result[1]) = convert_bfloat16_float(src[0]);
|
| /external/pytorch/aten/src/ATen/cpu/vec/ |
| D | functional_bfloat16.h | 26 return convert_bfloat16_float(a);
|