提交 02c61708 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Tag safe ops.

上级 4979f7e0
...@@ -602,6 +602,7 @@ class Rebroadcast(gof.Op): ...@@ -602,6 +602,7 @@ class Rebroadcast(gof.Op):
..note: works inplace and works for CudaNdarrayType ..note: works inplace and works for CudaNdarrayType
""" """
view_map = {0: [0]} view_map = {0: [0]}
_f16_ok = True
# Mapping from Type to C code (and version) to use. # Mapping from Type to C code (and version) to use.
# In the C code, the name of the input variable is %(iname)s, # In the C code, the name of the input variable is %(iname)s,
# the output variable is %(oname)s. # the output variable is %(oname)s.
......
...@@ -3858,6 +3858,7 @@ class Reshape(Op): ...@@ -3858,6 +3858,7 @@ class Reshape(Op):
The number of dimensions to which to reshape to (ndim) must be The number of dimensions to which to reshape to (ndim) must be
known at graph build time.""" known at graph build time."""
view_map = {0: [0]} # output 0 is potentially aliased to inputs [0] view_map = {0: [0]} # output 0 is potentially aliased to inputs [0]
_f16_ok = True
check_input = False check_input = False
......
...@@ -291,6 +291,7 @@ class Subtensor(Op): ...@@ -291,6 +291,7 @@ class Subtensor(Op):
debug = 0 debug = 0
check_input = False check_input = False
view_map = {0: [0]} view_map = {0: [0]}
_f16_ok = True
@staticmethod @staticmethod
def collapse(idxs, cond): def collapse(idxs, cond):
...@@ -1603,6 +1604,7 @@ class AdvancedSubtensor1(Op): ...@@ -1603,6 +1604,7 @@ class AdvancedSubtensor1(Op):
# sparse_grad doesn't go in here since it only affects the output # sparse_grad doesn't go in here since it only affects the output
# of the grad() method. # of the grad() method.
__props__ = () __props__ = ()
_f16_ok = True
def __init__(self, sparse_grad=False): def __init__(self, sparse_grad=False):
self.sparse_grad = sparse_grad self.sparse_grad = sparse_grad
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论