提交 875d0364 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

import cuda in perform for allocating zeros on cuda

This import used to be on top of the file, but that creates a cycling import.
上级 21fac54d
...@@ -586,6 +586,8 @@ class Scan(Op): ...@@ -586,6 +586,8 @@ class Scan(Op):
Y sequence outputs y_1, y_2, ... y_<self.n_outs> Y sequence outputs y_1, y_2, ... y_<self.n_outs>
""" """
# In order to be able to allocate cuda ndarrays if needed
from theano.sandbox import cuda
# 1. Unzip the number of steps and sequences. If number of steps is # 1. Unzip the number of steps and sequences. If number of steps is
# negative flip sequences around, and make n_steps positive # negative flip sequences around, and make n_steps positive
t0_call = time.time() t0_call = time.time()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论