FM / cpu

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu/

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu

Name	Size	Mode	Actions
conv_packed_params.h	791	0644	edit dl rm
conv_serialization.h	11455	0644	edit dl rm
embedding_packed_params.h	888	0644	edit dl rm
fbgemm_utils.h	10887	0644	edit dl rm
init_qnnpack.h	146	0644	edit dl rm
packed_params.h	1467	0644	edit dl rm
qembeddingbag.h	895	0644	edit dl rm
qembeddingbag_prepack.h	236	0644	edit dl rm
qnnpack_utils.h	15853	0644	edit dl rm
quantized_ops.h	5860	0644	edit dl rm
quant_utils.h	7383	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu/packed_params.h (1467B)

#pragma once

#include <ATen/core/ivalue.h>

struct LinearPackedParamsBase : public torch::jit::CustomClassHolder {
  virtual at::Tensor apply(
      at::Tensor input,
      double output_scale,
      int64_t output_zero_point) = 0;
  virtual at::Tensor apply_relu(
      at::Tensor input,
      double output_scale,
      int64_t output_zero_point) = 0;

// out variant of LinearPackedParamsBase::apply
  virtual at::Tensor& apply_out(
      const at::Tensor& input,
      double output_scale,
      int64_t output_zero_point,
      at::Tensor& output) {
    throw std::runtime_error(
        "apply_out is not implemented for this packed "
        "parameter type");
    return output;
  }

virtual at::Tensor& apply_relu_out(
      const at::Tensor& input,
      double output_scale,
      int64_t output_zero_point,
      at::Tensor& output) {
    throw std::runtime_error(
        "apply_relu_out is not implemented for this packed "
        "parameter type");
    return output;
  }

virtual at::Tensor apply_dynamic(at::Tensor input, bool reduce_range=false) = 0;
  virtual at::Tensor apply_dynamic_relu(at::Tensor input, bool reduce_range=false) = 0;

virtual std::tuple<at::Tensor, c10::optional<at::Tensor>> unpack() = 0;

virtual c10::optional<at::Tensor> bias() = 0;

virtual void set_bias(c10::optional<at::Tensor> bias) {
    throw std::runtime_error(
        "set_bias is not implemented for this packed "
        "parameter type");
  }
};

#pragma once #include struct LinearPackedParamsBase : public torch::jit::CustomClassHolder { virtual at::Tensor apply( at::Tensor input, double output_scale, int64_t output_zero_point) = 0; virtual at::Tensor apply_relu( at::Tensor input, double output_scale, int64_t output_zero_point) = 0; // out variant of LinearPackedParamsBase::apply virtual at::Tensor& apply_out( const at::Tensor& input, double output_scale, int64_t output_zero_point, at::Tensor& output) { throw std::runtime_error( "apply_out is not implemented for this packed " "parameter type"); return output; } virtual at::Tensor& apply_relu_out( const at::Tensor& input, double output_scale, int64_t output_zero_point, at::Tensor& output) { throw std::runtime_error( "apply_relu_out is not implemented for this packed " "parameter type"); return output; } virtual at::Tensor apply_dynamic(at::Tensor input, bool reduce_range=false) = 0; virtual at::Tensor apply_dynamic_relu(at::Tensor input, bool reduce_range=false) = 0; virtual std::tuple> unpack() = 0; virtual c10::optional bias() = 0; virtual void set_bias(c10::optional bias) { throw std::runtime_error( "set_bias is not implemented for this packed " "parameter type"); } };