提交 edbe833a authored 作者: Frederic's avatar Frederic

Fix a crash in new tests as we didn't converted to a TensorVariable.

上级 99fa5f39
...@@ -1546,6 +1546,7 @@ class CAReduceDtype(CAReduce): ...@@ -1546,6 +1546,7 @@ class CAReduceDtype(CAReduce):
# We need to redefine make_node so that, if self.dtype is None, # We need to redefine make_node so that, if self.dtype is None,
# we can infer what dtype should be, and create a node from an Op # we can infer what dtype should be, and create a node from an Op
# of the appropriate dtype. # of the appropriate dtype.
input = as_tensor_variable(input)
dtype = self._output_dtype(input.dtype) dtype = self._output_dtype(input.dtype)
assert dtype is not None assert dtype is not None
if dtype == self.dtype: if dtype == self.dtype:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论