提交 f28d5cb9 authored 作者: James Bergstra's avatar James Bergstra

added dot and gemm-related optimizations to FAST_RUN from autoencoder/aa.py

上级 a489cb3e
......@@ -205,10 +205,8 @@ class DimShuffle(Op):
if self.inplace:
print "INPLACE"
get_base = ['{ PyArrayObject * base = %(input)s', 'Py_INCREF((PyObject*)base)']
else:
print "NOT INPLACE"
get_base = [('{ PyArrayObject * base = (PyArrayObject*)PyArray_FromAny((PyObject*)%(input)s, NULL,'
'0, 0, NPY_ALIGNED|NPY_ENSURECOPY, NULL)')]
......
差异被折叠。
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论