提交 f02f0756 authored 作者: notoraptor's avatar notoraptor

Assert first node is a TensorType.

上级 927aac6a
...@@ -475,7 +475,7 @@ class mrg_uniform(mrg_uniform_base): ...@@ -475,7 +475,7 @@ class mrg_uniform(mrg_uniform_base):
# If we try to use the C code here with something else than a # If we try to use the C code here with something else than a
# TensorType, something is wrong (likely one of the GPU ops # TensorType, something is wrong (likely one of the GPU ops
# not defining C code correctly). # not defining C code correctly).
# NB: first node should always be a TensorType, as it is created with as_tensor_variable() into make_node. assert isinstance(node.inputs[0].type, TensorType)
return """ return """
//////// <code generated by mrg_uniform> //////// <code generated by mrg_uniform>
npy_int64 odims_i; npy_int64 odims_i;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论