Searched refs:tuple_index_ (Results 1 – 5 of 5) sorted by relevance
118 if (instruction->tuple_index() != tuple_index_) { in MatchAndExplain()120 << ", want " << tuple_index_ << ")"; in MatchAndExplain()
79 tuple_index_(tuple_index) {}85 int64 tuple_index_;
1540 : tuple_index_(tuple_index) {}1551 *os << "which is a GTE with index " << tuple_index_;1558 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_1562 if (inst->tuple_index() != tuple_index_) {1563 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_;1569 int64 tuple_index_;
1101 int64 tuple_index() const { return tuple_index_; } in tuple_index()1104 tuple_index_ = new_tuple_index; in set_tuple_index()1121 int64 tuple_index_ = -1; variable
1972 : HloInstruction(HloOpcode::kGetTupleElement, shape), tuple_index_(index) { in HloGetTupleElementInstruction()1978 proto.set_tuple_index(tuple_index_); in ToProto()