File tree Expand file tree Collapse file tree 1 file changed +39
-0
lines changed
backends/cadence/hifi/operators Expand file tree Collapse file tree 1 file changed +39
-0
lines changed Original file line number Diff line number Diff line change 12
12
_ (uint8_t , Byte) \
13
13
_(int8_t , Char)
14
14
15
+ using ::executorch::aten::IntArrayRef;
15
16
using ::executorch::aten::optional;
16
17
using ::executorch::aten::ScalarType;
17
18
using ::executorch::aten::Tensor;
@@ -67,6 +68,44 @@ void quantized_linear_per_tensor_out(
67
68
__ET_UNUSED const optional<Tensor>& offset,
68
69
Tensor& out);
69
70
71
+ void quantized_conv_out (
72
+ __ET_UNUSED KernelRuntimeContext& ctx,
73
+ const Tensor& input,
74
+ const Tensor& weight,
75
+ const Tensor& bias,
76
+ IntArrayRef stride,
77
+ IntArrayRef padding,
78
+ IntArrayRef dilation,
79
+ int64_t groups,
80
+ int64_t in_zero_point,
81
+ const Tensor& weight_zero_point,
82
+ const Tensor& bias_scale,
83
+ double output_scale,
84
+ int64_t output_zero_point,
85
+ __ET_UNUSED const Tensor& out_multiplier,
86
+ __ET_UNUSED const Tensor& out_shift,
87
+ bool channel_last,
88
+ Tensor& out);
89
+
90
+ void quantized_conv_per_tensor_out (
91
+ __ET_UNUSED KernelRuntimeContext& ctx,
92
+ const Tensor& input,
93
+ const Tensor& weight,
94
+ const Tensor& bias,
95
+ IntArrayRef stride,
96
+ IntArrayRef padding,
97
+ IntArrayRef dilation,
98
+ int64_t groups,
99
+ int64_t in_zero_point,
100
+ int64_t weight_zero_point,
101
+ double bias_scale,
102
+ double output_scale,
103
+ int64_t output_zero_point,
104
+ __ET_UNUSED int64_t out_multiplier,
105
+ __ET_UNUSED int64_t out_shift,
106
+ bool channel_last,
107
+ Tensor& out);
108
+
70
109
} // namespace native
71
110
} // namespace HiFi
72
111
} // namespace impl
You can’t perform that action at this time.
0 commit comments