Searched refs:GetAllTuples (Results 1 – 2 of 2) sorted by relevance
220 absl::optional<std::vector<HloInstruction*>> ArCrsCombiner::GetAllTuples( in GetAllTuples() function in xla::ArCrsCombiner233 return GetAllTuples(instruction->operands()[0], visited); in GetAllTuples()239 auto init_tuples = GetAllTuples(while_instr->while_init(), visited); in GetAllTuples()240 auto body_tuples = GetAllTuples( in GetAllTuples()260 GetAllTuples(cond_instr->mutable_operand(i + 1), visited); in GetAllTuples()274 auto tuples = GetAllTuples(instruction->operands()[0], visited); in GetAllTuples()279 auto tmp_tuples = GetAllTuples( in GetAllTuples()300 auto init_tuples = GetAllTuples(instruction->while_init(), visited); in GetAllTuples()302 GetAllTuples(instruction->while_body()->root_instruction(), visited); in GetAllTuples()319 auto tuples = GetAllTuples(tuple_shaped_instruction, &visited); in TupleElementsComputeSameValue()
138 absl::optional<std::vector<HloInstruction*>> GetAllTuples(