提交 2479cc7e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add __props__ to blocksparse ops.

上级 0900b662
......@@ -22,6 +22,7 @@ class SparseBlockGemv(Op):
:scale: 50 %
"""
__props__ = ('inplace',)
registered_opts = []
......@@ -138,6 +139,7 @@ class SparseBlockOuter(Op):
This op is involved in the gradient of SparseBlockGemv.
"""
__props__ = ('inplace',)
registered_opts = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论