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

Tag safe ops.

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