Method encode

src/ailego/algorithm/binary_quantizer.cc:40–57

Quantize data: encode the float input to uint32_t output

//! Quantize data: encode the float input to uint32_t output
void BinaryQuantizer::encode(const float *in, size_t dim, uint32_t *out) const {
  for (size_t i = 0; i < dim; i += 32) {
    size_t remain = i + 32 <= dim ? 32 : dim - i;
    uint32_t data = 0;
    uint32_t mask = 1;

    for (size_t j = 0; j < remain; j++) {
      if (in[i + j] >= threshold_) {
        data |= mask;
      }

      mask <<= 1;
    }

    *out = data;
    out++;
  }
}

//! De-quantize data: decode the input uint32_t to float output
//!   bit value 1 will be mapped to 1.0