Searched refs:branch_graphs (Results 1 – 2 of 2) sorted by relevance
467 def _make_intermediates_match(branch_graphs, branch_optionals): argument487 for i, branch_graph in enumerate(branch_graphs):494 def _make_intermediates_match_xla(branch_graphs, branch_intermediates): argument497 for i, branch_graph in enumerate(branch_graphs):509 def _make_inputs_match(branch_graphs, branch_inputs): argument526 assert len(branch_graphs) == len(branch_inputs)536 for branch_graph, branch_in in zip(branch_graphs, branch_inputs):582 def _make_output_composite_tensors_match(op_type, branch_graphs): argument600 assert branch_graphs601 branch_outputs = [g.structured_outputs for g in branch_graphs][all …]
1387 def _set_read_only_resource_inputs_attr(op, branch_graphs): argument1397 for branch_graph in branch_graphs: