提交 2e9c2df1 authored 作者: Saizheng Zhang's avatar Saizheng Zhang

fix import theano in gpuarray/basic_op.py and scanmodule

上级 1c6b1d1b
......@@ -4,6 +4,7 @@ import copy
import re
import numpy as np
import theano
from theano import Op, Apply, Type, Variable
from theano import tensor, config
from theano.gradient import grad_undefined
......
......@@ -2867,8 +2867,7 @@ class Scan(PureOp):
gof.ops_with_inner_function[Scan] = 'fn'
# TODO: move that to the new back-end and new profiling.py print_tips
# @theano.compile.profiling.register_profiler_printer
@theano.compile.profiling.register_profiler_printer
def profile_printer(fct_name, compile_time, fct_call_time, fct_call,
apply_time, apply_cimpl, message, outputs_size,
other_time):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论