Searched refs:branch_count (Results 1 – 18 of 18) sorted by relevance
55 if (conditional->branch_count() > 1) { in TryRemoveConditional()67 if (branch_index < 0 || branch_index >= conditional->branch_count()) { in TryRemoveConditional()68 branch_index = conditional->branch_count() - 1; in TryRemoveConditional()
52 for (int b = 0; b < instruction->branch_count(); ++b) { in ReplaceCalledComputation()53 if (b == instruction->branch_count() - 1) { in ReplaceCalledComputation()
78 for (int j = 0; j < a_ancestor->branch_count(); ++j) { in ExecutesBefore()158 for (int j = 0; j < b.defining_instruction()->branch_count(); ++j) { in IsDefinedBefore()235 for (int j = 0; j < conditional->branch_count(); ++j) { in UseIsBeforeValueDefinition()
377 std::vector<const InstructionValueSet*> inputs(conditional->branch_count()); in UpdateConditionalValueSet()378 for (int j = 0; j < conditional->branch_count(); ++j) { in UpdateConditionalValueSet()517 for (int j = 0; j < conditional->branch_count(); ++j) { in UpdateParameterValueSet()682 for (int j = 0; j < user->branch_count(); ++j) { in Propagate()
596 for (int j = 1; j < instruction->branch_count(); ++j) { in AddMandatoryConstraints()607 for (int k = 0; k < instruction->branch_count(); ++k) { in AddMandatoryConstraints()609 int j = (k + largest_branch) % instruction->branch_count(); in AddMandatoryConstraints()710 for (int j = 0; j < instruction->branch_count(); ++j) { in CheckConditionalLayout()
307 for (int b = 0; b < hlo->branch_count(); ++b) { in RecordInstructions()
670 const int num_branches = conditional->branch_count(); in HandleConditional()1375 for (int b = 0; b < conditional->branch_count(); ++b) { in HandleConditional()
161 for (int b = 0; b < instruction->branch_count(); ++b) { in ReplaceComputations()
693 for (int j = 1; j < conditional->branch_count(); ++j) { in HandleConditional()
1632 for (int j = 0; j < conditional->branch_count(); ++j) { in BuildColocatedBufferSets()1641 for (int j = 0; j < conditional->branch_count(); ++j) { in BuildColocatedBufferSets()
1269 if (branch_index < 0 || branch_index >= conditional->branch_count()) { in HandleConditional()1270 branch_index = conditional->branch_count() - 1; in HandleConditional()
1478 CHECK_EQ(new_operands.size(), branch_count() + 1); in CloneWithNewOperands()1786 for (int j = 0; j < branch_count(); ++j) { in IdenticalSlowPath()2075 int HloInstruction::branch_count() const { in branch_count() function in xla::HloInstruction
1081 int branch_count() const;
84 if (branch_index < 0 || branch_index >= hlo_instruction()->branch_count()) { in ExecuteOnStream()85 branch_index = hlo_instruction()->branch_count() - 1; in ExecuteOnStream()
2096 for (int j = 0; j < conditional->branch_count(); ++j) { in CheckConditionalBuffersShareAllocation()2160 for (int j = 0; j < hlo->branch_count(); ++j) { in BuildConditionalThunk()
117 for name, branch_count in iteritems(stores):118 if branch_count == len(branch_symbols):
4507 for (int branch_count = 1; branch_count < kMaxBranchCount; branch_count++) { in TEST_T32() local4508 Label* targets = new Label[branch_count]; in TEST_T32()4510 for (int i = 0; i < branch_count; i++) { in TEST_T32()4514 for (int i = 0; i < branch_count; i++) { in TEST_T32()4515 __ Bind(&targets[branch_count - i - 1]); in TEST_T32()
2533 int num_branches = conditional->branch_count(); in HandleConditional()