add input information into LiteNetworkIO
(
self, obj, is_host=True, io_type=LiteIOType.LITE_IO_VALUE, layout=None
)
| 421 | self.outputs.append(i) |
| 422 | |
| 423 | def add_input( |
| 424 | self, obj, is_host=True, io_type=LiteIOType.LITE_IO_VALUE, layout=None |
| 425 | ): |
| 426 | """ |
| 427 | add input information into LiteNetworkIO |
| 428 | """ |
| 429 | if isinstance(obj, LiteIO): |
| 430 | self.inputs.append(obj) |
| 431 | else: |
| 432 | name = obj |
| 433 | self.add_input(LiteIO(name, is_host, io_type, layout)) |
| 434 | |
| 435 | def add_output( |
| 436 | self, obj, is_host=True, io_type=LiteIOType.LITE_IO_VALUE, layout=None |