Home
last modified time | relevance | path

Searched refs:tuple_index_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.cc114 if (instruction->tuple_index() != tuple_index_) { in MatchAndExplain()
116 << ", want " << tuple_index_ << ")"; in MatchAndExplain()
Dhlo_matchers.h79 tuple_index_(tuple_index) {}
85 int64 tuple_index_;
Dpattern_matcher.h1554 : tuple_index_(tuple_index) {}
1565 *os << "which is a GTE with index " << tuple_index_;
1572 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_
1576 if (inst->tuple_index() != tuple_index_) {
1577 EXPLAIN << "HloInstruction is not a GTE with index " << tuple_index_;
1583 int64 tuple_index_;
Dhlo_instructions.h913 int64 tuple_index() const { return tuple_index_; } in tuple_index()
929 int64 tuple_index_ = -1; variable
Dhlo_instructions.cc1697 : HloInstruction(HloOpcode::kGetTupleElement, shape), tuple_index_(index) { in HloGetTupleElementInstruction()
1703 proto.set_tuple_index(tuple_index_); in ToProto()