saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix error message when running NhwcConv with a bad weight channel count (#15221)

Browse source
This commit is contained in:
Patrice Vignola 2023-03-27 00:15:19 -07:00 committed by GitHub
parent d182d34f1d
commit b10aaf4e9c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/core/providers/cpu/nn/conv_attributes.h
View file

@ -113,11 +113,12 @@ struct ConvAttributes {
const int64_t M = weight_shape[0];
const int64_t C = input_channels_last ? input_shape.GetDims().back() : input_shape[1];
const int64_t weight_channels = weight_channels_last ? weight_shape.GetDims().back() : weight_shape[1];
if (C != (weight_channels_last ? weight_shape.GetDims().back() : weight_shape[1]) * group) {
if (C != weight_channels * group) {
return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, "Input channels C is not equal to kernel channels * group.",
" C: ", C,
" kernel channels: ", weight_shape[1],
" kernel channels: ", weight_channels,
" group: ", group);
}