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