提交 5e9abd7c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Tag Alloc and Assert for float16 since they will work.

上级 9c91657a
......@@ -2774,6 +2774,7 @@ class Alloc(gof.Op):
are lifted, the first argument to fill can often be pruned from the graph.
"""
_f16_ok = True
__props__ = ()
def validate_shape(self, shape):
......
......@@ -2238,7 +2238,7 @@ class Assert(T.Op):
>>> func = theano.function([x], assert_op(x, x.size<2))
"""
_f16_ok = True
__props__ = ('msg',)
view_map = {0: [0]}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论