Searched refs:branch_graphs (Results 1 – 2 of 2) sorted by relevance
504 def _make_intermediates_match(branch_graphs, branch_optionals): argument524 for i, branch_graph in enumerate(branch_graphs):531 def _make_intermediates_match_xla(branch_graphs, branch_intermediates): argument534 for i, branch_graph in enumerate(branch_graphs):546 def _make_inputs_match(branch_graphs, branch_inputs): argument563 assert len(branch_graphs) == len(branch_inputs)573 for branch_graph, branch_in in zip(branch_graphs, branch_inputs):619 def _make_output_composite_tensors_match(op_type, branch_graphs): argument637 assert branch_graphs638 branch_outputs = [g.structured_outputs for g in branch_graphs][all …]
1449 def _set_read_only_resource_inputs_attr(op, branch_graphs): argument1459 for branch_graph in branch_graphs: