Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 28 additions & 0 deletions backends/cortex_m/ops/op_quantized_avg_pool2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -184,5 +184,33 @@ Tensor& quantized_avg_pool2d_out(
out);
}

// cppcheck-suppress unusedFunction
Tensor& quantized_avg_pool2d_nhwc_out(
KernelRuntimeContext& context,
const Tensor& input,
const Int64ArrayRef kernel_size,
const Int64ArrayRef stride,
const Int64ArrayRef padding,
const bool ceil_mode,
const int64_t zero_point,
const int64_t multiplier,
const int64_t shift,
const Tensor& scratch,
Tensor& out) {
return quantized_avg_pool2d_out_impl(
context,
input,
kernel_size,
stride,
padding,
ceil_mode,
zero_point,
multiplier,
shift,
scratch,
ActivationLayout::NHWCLogical,
out);
}

} // namespace native
} // namespace cortex_m
36 changes: 36 additions & 0 deletions backends/cortex_m/ops/op_quantized_conv2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -280,5 +280,41 @@ Tensor& quantized_conv2d_out(
out);
}

// cppcheck-suppress unusedFunction
Tensor& quantized_conv2d_nhwc_out(
KernelRuntimeContext& context,
const Tensor& input,
const Tensor& weight,
const std::optional<Tensor>& bias,
const Int64ArrayRef stride,
const Int64ArrayRef padding,
const Int64ArrayRef dilation,
const int64_t input_offset,
const int64_t output_offset,
const Tensor& requantize_multipliers,
const Tensor& requantize_shifts,
const int64_t activation_min,
const int64_t activation_max,
const Tensor& scratch,
Tensor& out) {
return quantized_conv2d_out_impl(
context,
input,
weight,
bias,
stride,
padding,
dilation,
input_offset,
output_offset,
requantize_multipliers,
requantize_shifts,
activation_min,
activation_max,
scratch,
ActivationLayout::NHWCLogical,
out);
}

} // namespace native
} // namespace cortex_m
38 changes: 38 additions & 0 deletions backends/cortex_m/ops/op_quantized_depthwise_conv2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -331,5 +331,43 @@ Tensor& quantized_depthwise_conv2d_out(
out);
}

// cppcheck-suppress unusedFunction
Tensor& quantized_depthwise_conv2d_nhwc_out(
KernelRuntimeContext& context,
const Tensor& input,
const Tensor& weight,
const std::optional<Tensor>& bias,
const Int64ArrayRef stride,
const Int64ArrayRef padding,
const Int64ArrayRef dilation,
const int64_t depth_multiplier,
const int64_t input_offset,
const int64_t output_offset,
const Tensor& requantize_multipliers,
const Tensor& requantize_shifts,
const int64_t activation_min,
const int64_t activation_max,
const Tensor& scratch,
Tensor& out) {
return quantized_depthwise_conv2d_out_impl(
context,
input,
weight,
bias,
stride,
padding,
dilation,
depth_multiplier,
input_offset,
output_offset,
requantize_multipliers,
requantize_shifts,
activation_min,
activation_max,
scratch,
ActivationLayout::NHWCLogical,
out);
}

} // namespace native
} // namespace cortex_m
30 changes: 30 additions & 0 deletions backends/cortex_m/ops/op_quantized_max_pool2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -129,5 +129,35 @@ Tensor& quantized_max_pool2d_out(
out);
}

// cppcheck-suppress unusedFunction
Tensor& quantized_max_pool2d_nhwc_out(
KernelRuntimeContext& context,
const Tensor& input,
const Int64ArrayRef kernel_size,
const Int64ArrayRef stride,
const Int64ArrayRef padding,
const Int64ArrayRef dilation,
const bool ceil_mode,
const int64_t input_zero_point,
const int64_t output_zero_point,
const int64_t activation_min,
const int64_t activation_max,
Tensor& out) {
return quantized_max_pool2d_out_impl(
context,
input,
kernel_size,
stride,
padding,
dilation,
ceil_mode,
input_zero_point,
output_zero_point,
activation_min,
activation_max,
ActivationLayout::NHWCLogical,
out);
}

} // namespace native
} // namespace cortex_m
40 changes: 40 additions & 0 deletions backends/cortex_m/ops/op_quantized_transpose_conv2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -302,5 +302,45 @@ Tensor& quantized_transpose_conv2d_out(
out);
}

// cppcheck-suppress unusedFunction
Tensor& quantized_transpose_conv2d_nhwc_out(
KernelRuntimeContext& context,
const Tensor& input,
const Tensor& weight,
const std::optional<Tensor>& bias,
const Int64ArrayRef stride,
const Int64ArrayRef padding,
const Int64ArrayRef output_padding,
const Int64ArrayRef dilation,
const int64_t input_offset,
const int64_t output_offset,
const Tensor& requantize_multipliers,
const Tensor& requantize_shifts,
const int64_t activation_min,
const int64_t activation_max,
const Tensor& scratch,
const Tensor& output_scratch,
Tensor& out) {
return quantized_transpose_conv2d_out_impl(
context,
input,
weight,
bias,
stride,
padding,
output_padding,
dilation,
input_offset,
output_offset,
requantize_multipliers,
requantize_shifts,
activation_min,
activation_max,
scratch,
output_scratch,
ActivationLayout::NHWCLogical,
out);
}

} // namespace native
} // namespace cortex_m
Loading
Loading