Searched refs:gte_operand (Results 1 – 2 of 2) sorted by relevance
53 const HloInstruction* gte_operand) { in GetGTEOperandIndex() argument55 << gte_operand->ToString() << ")"; in GetGTEOperandIndex()73 m::GetTupleElement(m::Op().Is(gte_operand), tuple_idx))) { in GetGTEOperandIndex()
102 const HloInstruction* gte_operand = hlo->operand(0); in IsPositive() local103 switch (gte_operand->opcode()) { in IsPositive()105 const auto& target = gte_operand->custom_call_target(); in IsPositive()