to_vec_float_helper Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h lines 1258–1282

  template <
      typename U = T,
      std::enable_if_t<std::is_same_v<U, uint8_t>, int> = 0>
  Vectorized<int32_t> to_vec_float_helper() const {
    int32_t values[8] = {
        _vec0[0],
        _vec0[1],
        _vec0[2],
        _vec0[3],
        _vec0[4],
        _vec0[5],
        _vec0[6],
        _vec0[7],
    };

    return Vectorized<int32_t>{
        values[0],
        values[1],
        values[2],
        values[3],
        values[4],
        values[5],
        values[6],
        values[7]};
  }