Prompt48 commited on
Commit
7fb9e17
·
verified ·
1 Parent(s): 387071e

Upload edit\Qwen3-TTS-test\.venv\Lib\site-packages\torch\include\ATen\native\quantized\cpu\qconv.h with huggingface_hub

Browse files
edit//Qwen3-TTS-test//.venv//Lib//site-packages//torch//include//ATen//native//quantized//cpu//qconv.h ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #pragma once
2
+ #include <ATen/Tensor.h>
3
+ #include <ATen/Config.h>
4
+
5
+ namespace at {
6
+ namespace native {
7
+
8
+ class QConvoneDNN final {
9
+ public:
10
+
11
+ C10_API static at::Tensor run_pointwise(
12
+ at::Tensor act, // contains quantized values but not QTensor
13
+ double act_scale,
14
+ int64_t act_zero_point,
15
+ at::Tensor weight, // contains quantized values but not QTensor
16
+ at::Tensor weight_scales,
17
+ at::Tensor weight_zero_points,
18
+ std::optional<at::Tensor> bias,
19
+ torch::List<int64_t> stride,
20
+ torch::List<int64_t> padding,
21
+ torch::List<int64_t> dilation,
22
+ int64_t groups,
23
+ double output_scale,
24
+ int64_t output_zero_point,
25
+ std::optional<c10::ScalarType> output_dtype,
26
+ std::string_view attr,
27
+ torch::List<std::optional<at::Scalar>> scalars,
28
+ std::optional<std::string_view> algorithm);
29
+
30
+ C10_API static at::Tensor run_pointwise_tensor(
31
+ at::Tensor act, // contains quantized values but not QTensor
32
+ at::Tensor act_scale,
33
+ at::Tensor act_zero_point,
34
+ at::Tensor weight, // contains quantized values but not QTensor
35
+ at::Tensor weight_scales,
36
+ at::Tensor weight_zero_points,
37
+ std::optional<at::Tensor> bias,
38
+ torch::List<int64_t> stride,
39
+ torch::List<int64_t> padding,
40
+ torch::List<int64_t> dilation,
41
+ int64_t groups,
42
+ double output_scale,
43
+ int64_t output_zero_point,
44
+ std::optional<c10::ScalarType> output_dtype,
45
+ std::string_view attr,
46
+ torch::List<std::optional<at::Scalar>> scalars,
47
+ std::optional<std::string_view> algorithm);
48
+
49
+ C10_API static at::Tensor run_pointwise_binary(
50
+ at::Tensor act, // contains quantized values but not QTensor
51
+ double act_scale,
52
+ int64_t act_zero_point,
53
+ at::Tensor weight, // contains quantized values but not QTensor
54
+ at::Tensor weight_scales,
55
+ at::Tensor weight_zero_points,
56
+ at::Tensor accum, // contains quantized values but not QTensor
57
+ std::optional<at::Tensor> bias,
58
+ torch::List<int64_t> stride,
59
+ torch::List<int64_t> padding,
60
+ torch::List<int64_t> dilation,
61
+ int64_t groups,
62
+ double output_scale,
63
+ int64_t output_zero_point,
64
+ std::optional<c10::ScalarType> output_dtype,
65
+ double accum_scale,
66
+ int64_t accum_zero_point,
67
+ std::string_view binary_attr,
68
+ std::optional<at::Scalar> alpha,
69
+ std::optional<std::string_view> unary_attr,
70
+ torch::List<std::optional<at::Scalar>> unary_scalars,
71
+ std::optional<std::string_view> unary_algorithm);
72
+
73
+ C10_API static at::Tensor run_pointwise_binary_tensor(
74
+ at::Tensor act, // contains quantized values but not QTensor
75
+ at::Tensor act_scale,
76
+ at::Tensor act_zero_point,
77
+ at::Tensor weight, // contains quantized values but not QTensor
78
+ at::Tensor weight_scales,
79
+ at::Tensor weight_zero_points,
80
+ at::Tensor accum, // contains quantized values but not QTensor
81
+ std::optional<at::Tensor> bias,
82
+ torch::List<int64_t> stride,
83
+ torch::List<int64_t> padding,
84
+ torch::List<int64_t> dilation,
85
+ int64_t groups,
86
+ double output_scale,
87
+ int64_t output_zero_point,
88
+ std::optional<c10::ScalarType> output_dtype,
89
+ double accum_scale,
90
+ int64_t accum_zero_point,
91
+ std::string_view binary_attr,
92
+ std::optional<at::Scalar> alpha,
93
+ std::optional<std::string_view> unary_attr,
94
+ torch::List<std::optional<at::Scalar>> unary_scalars,
95
+ std::optional<std::string_view> unary_algorithm);
96
+
97
+ };
98
+
99
+ } // namespace native
100
+ } // namespace at