Searched refs:HloDotInstruction (Results 1 – 6 of 6) sorted by relevance
104 std::unique_ptr<HloDotInstruction> CreateCanonicalDot(const Shape& shape,
575 std::unique_ptr<HloDotInstruction> CreateCanonicalDot(const Shape& shape, in CreateCanonicalDot()586 return absl::make_unique<HloDotInstruction>( in CreateCanonicalDot()
2509 HloDotInstruction::HloDotInstruction( in HloDotInstruction() function in xla::HloDotInstruction2520 HloInstructionProto HloDotInstruction::ToProto() const { in ToProto()2527 std::vector<string> HloDotInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2538 bool HloDotInstruction::IdenticalSlowPath( in IdenticalSlowPath()2542 const auto& casted_other = static_cast<const HloDotInstruction&>(other); in IdenticalSlowPath()2549 std::unique_ptr<HloInstruction> HloDotInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2553 return absl::make_unique<HloDotInstruction>( in CloneWithNewOperandsImpl()2558 string HloDotInstruction::DotDimensionNumbersToString() const { in DotDimensionNumbersToString()
561 instruction = absl::make_unique<HloDotInstruction>( in CreateFromProto()819 return absl::make_unique<HloDotInstruction>( in CreateDot()3180 if (auto* dot = DynCast<HloDotInstruction>(this)) { in precision_config()3190 if (auto* dot = DynCast<HloDotInstruction>(this)) { in mutable_precision_config()3530 return Cast<HloDotInstruction>(this)->dot_dimension_numbers(); in dot_dimension_numbers()
1460 class HloDotInstruction : public HloInstruction {1464 explicit HloDotInstruction(const Shape& shape, HloInstruction* lhs,
996 if (auto* dot_instr = DynCast<HloDotInstruction>(&instr)) { in IsBatchDot()