Browse Source

fix(imperative/src): fix adaptive_pooling bug

GitOrigin-RevId: e1dc46e3fa
release-1.10
Megvii Engine Team 3 years ago
parent
commit
c2eec47b42
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/src/impl/ops/adaptive_pooling.cpp

+ 1
- 0
imperative/src/impl/ops/adaptive_pooling.cpp View File

@@ -73,6 +73,7 @@ SmallVector<TensorPtr> apply_on_physical_tensor(
TensorLayout dst_layout = output_descs[0].layout;
if (!validated) {
TensorShape tshp;
dst_layout.ndim = src_layout.ndim;
dst_layout[0] = src_layout[0];
dst_layout[1] = src_layout[1];
if (inputs.size() == 2) {


Loading…
Cancel
Save