提交 1390deb4 authored 作者: Frederic's avatar Frederic

Test with clinker too as now the Shape Op have c code

上级 6f1c4d22
...@@ -410,9 +410,6 @@ class test_CAReduce(unittest_tools.InferShapeTester): ...@@ -410,9 +410,6 @@ class test_CAReduce(unittest_tools.InferShapeTester):
self.assertTrue((f_xv.shape == zv.shape), (f_xv, zv)) self.assertTrue((f_xv.shape == zv.shape), (f_xv, zv))
self.assertTrue(numpy.allclose(f_xv, zv), (f_xv, zv)) self.assertTrue(numpy.allclose(f_xv, zv), (f_xv, zv))
#test CAReduce.infer_shape
#the Shape op don't implement c_code!
if isinstance(linker, gof.PerformLinker):
x = TensorType(dtype, [(entry == 1) for entry in xsh])('x') x = TensorType(dtype, [(entry == 1) for entry in xsh])('x')
if tensor_op is None: if tensor_op is None:
e = self.op(scalar_op, axis=tosum)(x) e = self.op(scalar_op, axis=tosum)(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论