We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3c6fb60 commit a965531Copy full SHA for a965531
include/vpu/shave/VPUEM_piecewise_calc_subblk_size.h
@@ -81,7 +81,7 @@ class VPUEMCalcSubblk {
81
}
82
83
84
- if (m == 2) {
+ if (m == 3) {
85
blk_tensor.set_shape(m - 3, round_dim(max_num_data,
86
(blk_tensor.get_shape()[m - 1] * blk_tensor.get_shape()[m - 2])));
87
if (blk_tensor.get_shape()[m - 3] > int(ts.get_shape()[m - 2])) {
0 commit comments