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

fix test in FAST_COMPILE mode.

上级 4c9f0eaf
......@@ -64,9 +64,12 @@ def test_cdata():
i = TensorType('float32', (False,))()
c = ProdOp()(i)
i2 = GetOp()(c)
mode = None
if theano.config.mode == "FAST_COMPILE":
mode = "FAST_RUN"
# This should be a passthrough function for vectors
f = theano.function([i], i2)
f = theano.function([i], i2, mode=mode)
v = numpy.random.randn(9).astype('float32')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论