Python Variable.oの例

コード例 #1

ファイル: test_ir.py プロジェクト: leo-XUKANG/TensorRT-1

 def test_o_multiple_outputs(self):
     x = Variable(name="x")
     y = Variable(name="y")
     y2 = Variable(name="y2")
     node = Node(op="Add", name="Input", inputs=[x], outputs=[y])
     node2 = Node(op="Add", name="Input", inputs=[x], outputs=[y2])
     assert x.o() == y
     assert x.o(1) == y2

コード例 #2

ファイル: test_ir.py プロジェクト: leo-XUKANG/TensorRT-1

 def test_o(self):
     x = Variable(name="x")
     y = Variable(name="y")
     node = Node(op="Add", name="Input", inputs=[x], outputs=[y])
     assert x.o() == y