Searched refs:tuple_index_ (Results 1 – 5 of 5) sorted by relevance
120 if (instruction->tuple_index() != tuple_index_) { in MatchAndExplain()122 << ", want " << tuple_index_ << ")"; in MatchAndExplain()
82 tuple_index_(tuple_index) {}88 int64_t tuple_index_;
1654 : tuple_index_(tuple_index) {}1665 *os << "which is a GTE with index " << tuple_index_;1672 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_1676 if (inst->tuple_index() != tuple_index_) {1677 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_;1683 int64_t tuple_index_;
1451 int64_t tuple_index() const { return tuple_index_; } in tuple_index()1454 tuple_index_ = new_tuple_index; in set_tuple_index()1475 int64_t tuple_index_ = -1; variable
2268 : HloInstruction(HloOpcode::kGetTupleElement, shape), tuple_index_(index) { in HloGetTupleElementInstruction()2274 proto.set_tuple_index(tuple_index_); in ToProto()