(self, branch_a, branch_b, shared_input_branch=None,
shared_output_branch=None)
| 564 | """Multi IO Keras subclass model.""" |
| 565 | |
| 566 | def __init__(self, branch_a, branch_b, shared_input_branch=None, |
| 567 | shared_output_branch=None): |
| 568 | super(_MultiIOSubclassModel, self).__init__() |
| 569 | self._shared_input_branch = shared_input_branch |
| 570 | self._branch_a = branch_a |
| 571 | self._branch_b = branch_b |
| 572 | self._shared_output_branch = shared_output_branch |
| 573 | |
| 574 | def call(self, inputs, **kwargs): |
| 575 | if self._shared_input_branch: |