Searched refs:branch_intermediates (Results 1 – 1 of 1) sorted by relevance
531 def _make_intermediates_match_xla(branch_graphs, branch_intermediates): argument537 sum((bi for bi in branch_intermediates538 if bi is not branch_intermediates[i]), []))539 num_preceding = sum(len(bi) for bi in branch_intermediates[:i])541 branch_intermediates[i] +1109 branch_intermediates = [1114 branch_intermediates)