Skip to content

Commit 2be5be9

Browse files
author
Zonglin Peng
committed
add im2row and conv
1 parent 5fe0f1c commit 2be5be9

File tree

5 files changed

+875
-262
lines changed

5 files changed

+875
-262
lines changed

backends/cadence/aot/functions.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,3 +188,13 @@
188188
kernels:
189189
- arg_meta: null
190190
kernel_name: impl::reference::quantized_linear_per_tensor_out
191+
192+
- func: cadence::im2row.out(Tensor input, int[2] kernel_size, int[2] dilation, int[2] padding, int[2] stride, Tensor in_zero_point, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)
193+
kernels:
194+
- arg_meta: null
195+
kernel_name: impl::reference::im2row_out
196+
197+
- func: cadence::quantized_conv.per_tensor_out(Tensor input, Tensor weight, Tensor bias, int[] stride, SymInt[] padding, int[] dilation, int groups, int input_zero_point, int weight_zero_point, float bias_scale, float out_scale, int out_zero_point, int out_multiplier, int out_shift, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)
198+
kernels:
199+
- arg_meta: null
200+
kernel_name: impl::reference::quantized_conv_per_tensor_out

backends/cadence/reference/operators/CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,12 +73,13 @@ target_include_directories(
7373
add_library(
7474
custom_ops
7575
"quantized_linear_out.cpp"
76-
"quantized_conv_out.cpp"
7776
"quantized_relu_out.cpp"
7877
"quantized_layer_norm.cpp"
7978
"quantize_per_tensor.cpp"
8079
"dequantize_per_tensor.cpp"
8180
"quantized_matmul_out.cpp"
81+
"im2row_out.cpp"
82+
"convolution_out.cpp"
8283
)
8384
target_include_directories(
8485
custom_ops PUBLIC ${ROOT_DIR}/.. ${CMAKE_BINARY_DIR}

0 commit comments

Comments
 (0)