Searched defs:branch_b (Results 1 – 2 of 2) sorted by relevance
650 def __init__(self, branch_a, branch_b, shared_input_branch=None, argument733 branch_b, argument
3323 SpmdBuilder branch_b(absl::StrCat("infeed_branch_", i), visiting_hlo_); in HandleInfeed() local3753 SpmdBuilder branch_b(absl::StrCat("outfeed_branch_", i), visiting_hlo_); in HandleOutfeed() local