提交 35bf93a3 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

modifying tile test to test the Op

上级 5abbae65
...@@ -6879,31 +6879,33 @@ class TestInferShape(utt.InferShapeTester): ...@@ -6879,31 +6879,33 @@ class TestInferShape(utt.InferShapeTester):
# [Reshape(ndim)(adtens4, aivec)], # [Reshape(ndim)(adtens4, aivec)],
# [adtens4_val, [1, 3, 10, 4]], Reshape) # [adtens4_val, [1, 3, 10, 4]], Reshape)
# Tile op is depricated advec = dvector()
advec_val = rand(5)
# advec = dvector() aivec_val = [3]
# advec_val = rand(5) ndim = 1
# aivec_val = [3] self._compile_and_check([advec],
# ndim = 1 #[tile(advec, aivec_val, ndim)],
# self._compile_and_check([advec], [Tile(ndim)(advec, aivec_val)],
# [tile(advec, aivec_val, ndim)], [advec_val], Tile)
# [advec_val], Tile)
admat = dmatrix()
# admat = dmatrix() admat_val = rand(2, 4)
# admat_val = rand(2, 4) aivec_val = [2, 3]
# aivec_val = [2, 3] ndim = 2
# ndim = None self._compile_and_check([admat],
# self._compile_and_check([admat], #[tile(admat, aivec_val)],
# [tile(admat, aivec_val)], [Tile(ndim)(admat, aivec_val)],
# [admat_val], Tile) [admat_val], Tile)
# adtens4 = dtensor4() adtens4 = dtensor4()
# adtens4_val = rand(2, 4, 3, 5) adtens4_val = rand(2, 4, 3, 5)
# aivec_val = [2, 3, 1, 4] aivec_val = [2, 3, 1, 4]
# ndim = 4 ndim = 4
# self._compile_and_check([adtens4], self._compile_and_check([adtens4],
# [tile(adtens4, aivec_val, ndim)], #[tile(adtens4, aivec_val, ndim)],
# [adtens4_val], Tile) [Tile(ndim)(adtens4, aivec_val)],
[adtens4_val], Tile)
class TestTensorInstanceMethods(unittest.TestCase): class TestTensorInstanceMethods(unittest.TestCase):
def setUp(self): def setUp(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论