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

small fix from code review

上级 0ad8e57b
......@@ -352,7 +352,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
%(fail)s;
}
if (PyGpuArray_DIMS(%(sm)s)[0] !=
PyGpuArray__DIMS(%(y_idx)s)[0])
PyGpuArray_DIMS(%(y_idx)s)[0])
{
PyErr_SetString(PyExc_ValueError,
"sm.shape[0] != y_idx.shape[0]");
......
......@@ -7,7 +7,7 @@ from theano.tests.unittest_tools import SkipTest
from theano.tensor.tests import test_basic
import theano.sandbox.gpuarray
from theano.sandbox.gpuarray import basic_ops
from .. import basic_ops
from ..type import GpuArrayType, gpuarray_shared_constructor
from ..basic_ops import (GpuAlloc, GpuReshape, gpu_alloc,
gpu_from_host, host_from_gpu)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论