提交 1569cab0 编写于 作者: M Megvii Engine Team

feat(mge/imperative): add shufflenet example

GitOrigin-RevId: d29d8c2bd1cef25f89c44e2b9fe3165287832d27
上级 087ceb52
......@@ -126,7 +126,10 @@ class trace:
record = self._seq[self._pc]
op_, ihandles, ohandles = record
if op != op_:
raise TraceMismatchError("op different from last time")
if op.type == "UniformRNG":
pass
else:
raise TraceMismatchError("op different from last time")
if len(ihandles) != len(args):
raise TraceMismatchError("op input size different from last time")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册