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(rng): init layout strides
GitOrigin-RevId:
9833d866da
tags/v1.9.0
Megvii Engine Team
3 years ago
parent
9527859cc8
commit
1c1e9b002d
1 changed files
with
1 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-0
imperative/src/impl/ops/rng.cpp
+ 1
- 0
imperative/src/impl/ops/rng.cpp
View File
@@ -334,6 +334,7 @@ struct _InferLayout<true> {
for (size_t i = 0; i < target_ndim; ++i) {
out_layout.shape[i] = ptr[i];
}
out_layout.init_contiguous_stride();
return out_layout;
}
};
Write
Preview
Loading…
Cancel
Save