Searched refs:branch_args (Results 1 – 2 of 2) sorted by relevance
246 ValueRange branch_args, in ForwardCommonArgToOutput() argument283 result.replaceAllUsesWith(branch_args[common_arg_index.getValue()]); in ForwardCommonArgToOutput()292 ValueRange branch_args) { in CanonicalizeFunctionalIfCase() argument298 if (failed(ForwardCommonArgToOutput(op, branches, branch_args, in CanonicalizeFunctionalIfCase()
2864 std::vector<HloInstruction*> branch_args; in HandleConditional() local2872 branch_args.push_back(GetPartitionedHlo(hlo->operand(i + 1)).hlo()); in HandleConditional()2893 hlo->called_computations(), branch_args)); in HandleConditional()