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