提交 6545c206 authored 作者: Frederic's avatar Frederic

Fix compilation just introduced.

上级 2278e82d
......@@ -1441,7 +1441,7 @@ class MaxAndArgmax(Op):
else:
assert node.inputs[1].ndim == 0
axis_code = """
axis = ((%(dtype)s)PyArray_DATA(%(axis)s))[0];
axis = ((dtype_%(axis)s*)PyArray_DATA(%(axis)s))[0];
if(axis > PyArray_NDIM(%(x)s)-1 || axis < -PyArray_NDIM(%(x)s)){
PyErr_SetString(PyExc_ValueError, "MaxAndArgmax, bad axis argument");
%(fail)s
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论