Searched refs:HloCompareInstruction (Results 1 – 3 of 3) sorted by relevance
205 HloCompareInstruction::HloCompareInstruction(const Shape& shape, in HloCompareInstruction() function in xla::HloCompareInstruction214 HloInstructionProto HloCompareInstruction::ToProto() const { in ToProto()220 std::vector<string> HloCompareInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()225 bool HloCompareInstruction::IdenticalSlowPath( in IdenticalSlowPath()229 const auto& casted_other = static_cast<const HloCompareInstruction&>(other); in IdenticalSlowPath()233 std::unique_ptr<HloInstruction> HloCompareInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()237 return absl::make_unique<HloCompareInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
134 class HloCompareInstruction : public HloInstruction {136 explicit HloCompareInstruction(const Shape& shape, HloInstruction* lhs,
800 return absl::make_unique<HloCompareInstruction>(shape, lhs, rhs, direction); in CreateCompare()3542 return Cast<HloCompareInstruction>(this)->direction(); in comparison_direction()