提交 33d04174 authored 作者: Jesse Livezey's avatar Jesse Livezey

added mode

上级 9fb384e9
......@@ -283,7 +283,7 @@ class TestCorr2D(utt.InferShapeTester):
b_tens_val = rand(b_shape, dtype=b_dtype)
c_tens = op()(a_tens, b_tens)
f = theano.function([a_tens, b_tens], c_tens)
f = theano.function([a_tens, b_tens], c_tens, mode=self.mode)
assert_equals(f(a_tens_val, b_tens_val).dtype, c_dtype)
@attr('slow')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论