提交 cbee7123 authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

remove infershape cases that are no longer supported by Dot (scalar, tensor)

上级 f814c0c7
......@@ -6525,16 +6525,6 @@ class TestInferShape(utt.InferShapeTester):
# Dot
#scal/scal
adscal = dscalar()
bdscal = dscalar()
adscal_val = rand(1)[0]
bdscal_val = rand(1)[0]
self._compile_and_check([adscal, bdscal],
[Dot()(adscal, bdscal)],
[adscal_val, bdscal_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#vec/vec
advec = dvector()
bdvec = dvector()
......@@ -6555,40 +6545,6 @@ class TestInferShape(utt.InferShapeTester):
[admat_val, bdmat_val],
(Dot, tensor.blas.Dot22))
#tens3/tens3
adtens3 = dtensor3()
bdtens3 = dtensor3()
adtens3_val = rand(3, 4, 5)
bdtens3_val = rand(6, 5, 7)
self._compile_and_check([adtens3, bdtens3],
[Dot()(adtens3, bdtens3)],
[adtens3_val, bdtens3_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#scal/vec
self._compile_and_check([adscal, bdvec],
[Dot()(adscal, bdvec)],
[adscal_val, bdvec_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#scal/mat
self._compile_and_check([adscal, bdmat],
[Dot()(adscal, bdmat)],
[adscal_val, bdmat_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#scal/tens3
self._compile_and_check([adscal, bdtens3],
[Dot()(adscal, bdtens3)],
[adscal_val, bdtens3_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#vec/scal
self._compile_and_check([advec, bdscal],
[Dot()(advec, bdscal)],
[advec_val, bdscal_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#vec/mat
bdmat_val = rand(4, 5)
self._compile_and_check([advec, bdmat],
......@@ -6596,19 +6552,6 @@ class TestInferShape(utt.InferShapeTester):
[advec_val, bdmat_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#vec/tens3
bdtens3_val = rand(3, 4, 5)
self._compile_and_check([advec, bdtens3],
[Dot()(advec, bdtens3)],
[advec_val, bdtens3_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#mat/scal
self._compile_and_check([admat, bdscal],
[Dot()(admat, bdscal)],
[admat_val, bdscal_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#mat/vec
admat_val = rand(5, 4)
self._compile_and_check([admat, bdvec],
......@@ -6616,32 +6559,6 @@ class TestInferShape(utt.InferShapeTester):
[admat_val, bdvec_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#mat/tens3
self._compile_and_check([admat, bdtens3],
[Dot()(admat, bdtens3)],
[admat_val, bdtens3_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#tens3/scal
self._compile_and_check([adtens3, bdscal],
[Dot()(adtens3, bdscal)],
[adtens3_val, bdscal_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#tens3/vec
bdvec_val = rand(5)
self._compile_and_check([adtens3, bdvec],
[Dot()(adtens3, bdvec)],
[adtens3_val, bdvec_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
#tens3/mat
bdmat_val = rand(5, 4)
self._compile_and_check([adtens3, bdmat],
[Dot()(adtens3, bdmat)],
[adtens3_val, bdmat_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
# Split
aivec = ivector()
adtens_val = rand(4, 10, 3)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论