提交 0fae9d8b authored 作者: Frederic's avatar Frederic

Enable comment test with small fix to them.

上级 5d515fa9
...@@ -6448,8 +6448,6 @@ class TestInferShape(utt.InferShapeTester): ...@@ -6448,8 +6448,6 @@ class TestInferShape(utt.InferShapeTester):
[adtens3_val], Mean) [adtens3_val], Mean)
# IncSubtensor # IncSubtensor
# Note: Is testing only for the 4-tensor below sufficient?
# Please determine and take action.
admat = dmatrix() admat = dmatrix()
bdmat = dmatrix() bdmat = dmatrix()
advec = dvector() advec = dvector()
...@@ -6562,11 +6560,11 @@ class TestInferShape(utt.InferShapeTester): ...@@ -6562,11 +6560,11 @@ class TestInferShape(utt.InferShapeTester):
[adtens4_val, 1], [adtens4_val, 1],
AdvancedIncSubtensor1) AdvancedIncSubtensor1)
## TODO: (!!) function inc_subtensor fails on line 5784 in perform in basic.py
"""
aivec_val = [2, 3] aivec_val = [2, 3]
self._compile_and_check([admat, bdmat], self._compile_and_check([admat, bdmat],
[inc_subtensor(admat[aivec_val], bdmat)], [inc_subtensor(admat[aivec_val], bdmat)],
[admat_val, [[1, 2, 3, 4], [5, 6, 7, 8]]],
AdvancedIncSubtensor1)
aivec_val = [1, 3, 2] aivec_val = [1, 3, 2]
self._compile_and_check([admat, advec], self._compile_and_check([admat, advec],
...@@ -6583,22 +6581,22 @@ class TestInferShape(utt.InferShapeTester): ...@@ -6583,22 +6581,22 @@ class TestInferShape(utt.InferShapeTester):
aivec_val = [2, 3] aivec_val = [2, 3]
self._compile_and_check([adtens4, bdtens4], self._compile_and_check([adtens4, bdtens4],
[inc_subtensor(adtens4[aivec_val], bdtens4)], [inc_subtensor(adtens4[aivec_val], bdtens4)],
[adtens4_val, [[[[1, 2, 3, 4, 5]]]]], [adtens4_val, [[[[1, 2, 3, 4, 5]]],
[[[6, 7, 8, 9, 10]]]]],
AdvancedIncSubtensor1) AdvancedIncSubtensor1)
aivec_val = [1, 3, 2] aivec_val = [1, 2, 1]
self._compile_and_check([adtens4, advec], self._compile_and_check([adtens4, advec],
[inc_subtensor(adtens4[aivec_val], advec)], [inc_subtensor(adtens4[aivec_val], advec)],
[adtens4_val, [[1, 2, 3, 4, 5]]], [adtens4_val, [1, 2, 3, 4, 5]],
AdvancedIncSubtensor1) AdvancedIncSubtensor1)
aivec_val = [0, 3, 0] aivec_val = [0, 3, 0]
self._compile_and_check([adtens4, adscal], self._compile_and_check([adtens4, adscal],
[inc_subtensor(adtens4[aivec_val], adscal)], [inc_subtensor(adtens4[aivec_val], adscal)],
[adtens4_val, [[1, 2, 3, 4, 5]]], [adtens4_val, 2],
AdvancedIncSubtensor1) AdvancedIncSubtensor1)
"""
# AdvancedIncSubtensor # AdvancedIncSubtensor
# TODO: The shape is apparently generated correctly but the final # TODO: The shape is apparently generated correctly but the final
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论