QMulOnednn Class — pytorch Architecture

Source Code

aten/src/ATen/native/quantized/cpu/qmul.cpp lines 408–423

class QMulOnednn final {
  public:
  static Tensor run(
    const Tensor self, double self_scale, int64_t self_zero_point,
    const Tensor other, double other_scale, int64_t other_zero_point,
    double output_scale, int64_t output_zero_point, c10::ScalarType output_dtype
  ) {
#if AT_MKLDNN_ENABLED()
  return int8_mul_tensor_onednn(
    self, self_scale, self_zero_point,
    other, other_scale, other_zero_point,
    output_scale, output_zero_point, output_dtype);
#endif
  TORCH_CHECK(false, "Unimplemented (int8 mul tensor with onednn)");
  }
};