Searched refs:shared_input_branch (Results 1 – 1 of 1) sorted by relevance
442 def __init__(self, branch_a, branch_b, shared_input_branch=None, argument445 self._shared_input_branch = shared_input_branch523 shared_input_branch=None, argument603 if shared_input_branch:604 inputs = shared_input_branch[0]605 shared_input_branch = shared_input_branch[1:]613 return _MultiIOSubclassModel(branch_a, branch_b, shared_input_branch,619 (lambda: shared_input_branch),627 if shared_input_branch:629 for layer in shared_input_branch: