This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
fix(imperative): fix broadcast extra assert of output shape
GitOrigin-RevId:
e8dc7c4229
master
Megvii Engine Team
2 years ago
parent
7790040e23
commit
5c84305f19
1 changed files
with
0 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+0
-1
imperative/src/impl/ops/broadcast.cpp
+ 0
- 1
imperative/src/impl/ops/broadcast.cpp
View File
@@ -290,7 +290,6 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
}
}
if (op.axis != opr::Reshape::Param::INVALID_AXIS) {
mgb_assert(out_shape[op.axis] == -1);
out_shape[op.axis] = 1;
mgb_assert(
src.layout.total_nr_elems() % out_shape.total_nr_elems() == 0,
Write
Preview
Loading…
Cancel
Save