FM / quantized

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

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

Name	Size	Mode	Actions
cpu/	-	0755	rm
affine_quantizer.h	3622	0644	edit dl rm
affine_quantizer_base.h	1444	0644	edit dl rm
Copy.h	173	0644	edit dl rm
fake_quant_affine.h	1758	0644	edit dl rm

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

#pragma once

#include <ATen/ATen.h>
#include <ATen/native/DispatchStub.h>
#include <ATen/native/quantized/affine_quantizer_base.h>

namespace at {
namespace native {

Tensor& quantize_tensor_per_tensor_affine(
    const Tensor& rtensor,
    Tensor& qtensor,
    double scale,
    int64_t zero_point);
Tensor& quantize_tensor_per_channel_affine(
    const Tensor& rtensor,
    Tensor& qtensor,
    Tensor scales,
    Tensor zero_points,
    int64_t axis);

Tensor& quantize_tensor_per_channel_float_qparams(
    const Tensor& rtensor,
    Tensor& qtensor,
    Tensor scales,
    Tensor zero_points,
    int64_t axis);

Tensor& dequantize_tensor_per_tensor_affine(
    const Tensor& qtensor,
    Tensor& rtensor,
    double scale,
    int64_t zero_point);
Tensor& dequantize_tensor_per_channel_affine(
    const Tensor& qtensor,
    Tensor& rtensor,
    Tensor scales,
    Tensor zero_points,
    int64_t axis);
Tensor& dequantize_tensor_per_channel_float_qparams(
    const Tensor& qtensor,
    Tensor& rtensor,
    Tensor scales,
    Tensor zero_points,
    int64_t axis);

using quantize_tensor_per_tensor_affine_fn =
    void (*)(const Tensor& rtensor, Tensor& qtensor, double scale, int64_t zero_point);

using quantize_tensor_per_channel_affine_fn = void (*)(
    const Tensor& rtensor,
    Tensor& qtensor,
    const Tensor& scales,
    const Tensor& zero_points,
    int64_t axis);

using quantize_tensor_per_channel_float_qparams_fn = void (*)(
    const Tensor& rtensor,
    Tensor& qtensor,
    const Tensor& scales,
    const Tensor& zero_points,
    int64_t axis);

using dequantize_tensor_per_tensor_affine_fn =
    void (*)(const Tensor& qtensor, Tensor& rtensor, double scale, int64_t zero_point);

using dequantize_tensor_per_channel_affine_fn = void (*)(
    const Tensor& qtensor,
    Tensor& rtensor,
    const Tensor& scales,
    const Tensor& zero_points,
    int64_t axis);

using dequantize_tensor_per_channel_float_qparams_fn = void (*)(
    const Tensor& qtensor,
    Tensor& rtensor,
    const Tensor& scales,
    const Tensor& zero_points,
    int64_t axis);

using quantize_tensor_per_tensor_affine_sub_byte_fn =
    void (*)(const Tensor& rtensor, Tensor& qtensor, float scale, float zero_point);

using dequantize_tensor_per_tensor_affine_sub_byte_fn =
    void (*)(const Tensor& qtensor, Tensor& rtensor, float scale, float zero_point);

DECLARE_DISPATCH(
    quantize_tensor_per_tensor_affine_fn,
    quantize_tensor_per_tensor_affine_stub);
DECLARE_DISPATCH(
    quantize_tensor_per_channel_affine_fn,
    quantize_tensor_per_channel_affine_stub);
DECLARE_DISPATCH(
    quantize_tensor_per_channel_float_qparams_fn,
    quantize_tensor_per_channel_float_qparams_stub);

DECLARE_DISPATCH(
    dequantize_tensor_per_tensor_affine_fn,
    dequantize_tensor_per_tensor_affine_stub);
DECLARE_DISPATCH(
    dequantize_tensor_per_channel_affine_fn,
    dequantize_tensor_per_channel_affine_stub);
DECLARE_DISPATCH(
    dequantize_tensor_per_channel_float_qparams_fn,
    dequantize_tensor_per_channel_float_qparams_stub);

DECLARE_DISPATCH(
    quantize_tensor_per_tensor_affine_sub_byte_fn,
    quantize_tensor_per_tensor_affine_sub_byte_stub);

DECLARE_DISPATCH(
    dequantize_tensor_per_tensor_affine_sub_byte_fn,
    dequantize_tensor_per_tensor_affine_sub_byte_stub);

template <typename T>
TORCH_API Tensor quantize_tensor(
    Tensor rtensor,
    Tensor qtensor,
    double scale,
    int64_t zero_point);
template <typename T>
TORCH_API Tensor dequantize_tensor(
    Tensor qtensor,
    Tensor rtensor,
    double scale,
    int64_t zero_point);

} // namespace native
} // namespace at

#pragma once #include #include #include namespace at { namespace native { Tensor& quantize_tensor_per_tensor_affine( const Tensor& rtensor, Tensor& qtensor, double scale, int64_t zero_point); Tensor& quantize_tensor_per_channel_affine( const Tensor& rtensor, Tensor& qtensor, Tensor scales, Tensor zero_points, int64_t axis); Tensor& quantize_tensor_per_channel_float_qparams( const Tensor& rtensor, Tensor& qtensor, Tensor scales, Tensor zero_points, int64_t axis); Tensor& dequantize_tensor_per_tensor_affine( const Tensor& qtensor, Tensor& rtensor, double scale, int64_t zero_point); Tensor& dequantize_tensor_per_channel_affine( const Tensor& qtensor, Tensor& rtensor, Tensor scales, Tensor zero_points, int64_t axis); Tensor& dequantize_tensor_per_channel_float_qparams( const Tensor& qtensor, Tensor& rtensor, Tensor scales, Tensor zero_points, int64_t axis); using quantize_tensor_per_tensor_affine_fn = void (*)(const Tensor& rtensor, Tensor& qtensor, double scale, int64_t zero_point); using quantize_tensor_per_channel_affine_fn = void (*)( const Tensor& rtensor, Tensor& qtensor, const Tensor& scales, const Tensor& zero_points, int64_t axis); using quantize_tensor_per_channel_float_qparams_fn = void (*)( const Tensor& rtensor, Tensor& qtensor, const Tensor& scales, const Tensor& zero_points, int64_t axis); using dequantize_tensor_per_tensor_affine_fn = void (*)(const Tensor& qtensor, Tensor& rtensor, double scale, int64_t zero_point); using dequantize_tensor_per_channel_affine_fn = void (*)( const Tensor& qtensor, Tensor& rtensor, const Tensor& scales, const Tensor& zero_points, int64_t axis); using dequantize_tensor_per_channel_float_qparams_fn = void (*)( const Tensor& qtensor, Tensor& rtensor, const Tensor& scales, const Tensor& zero_points, int64_t axis); using quantize_tensor_per_tensor_affine_sub_byte_fn = void (*)(const Tensor& rtensor, Tensor& qtensor, float scale, float zero_point); using dequantize_tensor_per_tensor_affine_sub_byte_fn = void (*)(const Tensor& qtensor, Tensor& rtensor, float scale, float zero_point); DECLARE_DISPATCH( quantize_tensor_per_tensor_affine_fn, quantize_tensor_per_tensor_affine_stub); DECLARE_DISPATCH( quantize_tensor_per_channel_affine_fn, quantize_tensor_per_channel_affine_stub); DECLARE_DISPATCH( quantize_tensor_per_channel_float_qparams_fn, quantize_tensor_per_channel_float_qparams_stub); DECLARE_DISPATCH( dequantize_tensor_per_tensor_affine_fn, dequantize_tensor_per_tensor_affine_stub); DECLARE_DISPATCH( dequantize_tensor_per_channel_affine_fn, dequantize_tensor_per_channel_affine_stub); DECLARE_DISPATCH( dequantize_tensor_per_channel_float_qparams_fn, dequantize_tensor_per_channel_float_qparams_stub); DECLARE_DISPATCH( quantize_tensor_per_tensor_affine_sub_byte_fn, quantize_tensor_per_tensor_affine_sub_byte_stub); DECLARE_DISPATCH( dequantize_tensor_per_tensor_affine_sub_byte_fn, dequantize_tensor_per_tensor_affine_sub_byte_stub); template TORCH_API Tensor quantize_tensor( Tensor rtensor, Tensor qtensor, double scale, int64_t zero_point); template TORCH_API Tensor dequantize_tensor( Tensor qtensor, Tensor rtensor, double scale, int64_t zero_point); } // namespace native } // namespace at