提交 c8d6c1d0 编写于 作者: W whs 提交者: qingqing01

Auto generate python api for polygon_box_transform op. (#11074)

* Auto generate python api for polygon_box_transform op.

* Add unitest.
上级 5713a0bf
......@@ -71,6 +71,7 @@ __all__ = [
'cumsum',
'scatter',
'sum',
'polygon_box_transform',
'shape',
] + __activations__
......
......@@ -379,6 +379,14 @@ class TestBook(unittest.TestCase):
self.assertIsNotNone(output)
print(str(program))
def test_polygon_box_transform(self):
program = Program()
with program_guard(program):
x = layers.data(name='x', shape=[8, 4, 4], dtype="float32")
output = layers.polygon_box_transform(input=x)
self.assertIsNotNone(output)
print(str(program))
if __name__ == '__main__':
unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册