Searched refs:HloScatterInstruction (Results 1 – 13 of 13) sorted by relevance
2916 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2930 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2944 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2958 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2971 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2988 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()3006 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()3025 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()3045 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()3060 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()[all …]
180 const HloScatterInstruction& scatter);
2816 HloScatterInstruction::HloScatterInstruction( in HloScatterInstruction() function in xla::HloScatterInstruction2833 /*static*/ string HloScatterInstruction::ScatterDimensionNumbersToString( in ScatterDimensionNumbersToString()2855 HloScatterInstruction::MakeScatterDimNumbers( in MakeScatterDimNumbers()2875 HloInstructionProto HloScatterInstruction::ToProto() const { in ToProto()2883 std::vector<string> HloScatterInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2896 bool HloScatterInstruction::IdenticalSlowPath( in IdenticalSlowPath()2900 const auto& casted_other = static_cast<const HloScatterInstruction&>(other); in IdenticalSlowPath()2909 std::unique_ptr<HloInstruction> HloScatterInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2913 return absl::make_unique<HloScatterInstruction>( in CloneWithNewOperandsImpl()
1698 class HloScatterInstruction : public HloInstruction {1700 explicit HloScatterInstruction(
1058 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
1630 return absl::make_unique<HloScatterInstruction>( in CreateScatter()4246 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
1586 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()
1014 !DynCast<HloScatterInstruction>(lhs)->indices_are_sorted() && in HandleAdd()1015 !DynCast<HloScatterInstruction>(rhs)->indices_are_sorted(); in HandleAdd()
2516 HloScatterInstruction::MakeScatterDimNumbers( in ParseInstructionRhs()
594 Cast<const HloScatterInstruction>(use)->indices_are_sorted(); in CreateLiteralForConstrainedUses()
508 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter()
545 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in GetScatterDimensionNumbers()564 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in EmitScatterOp()
431 auto scatter = Cast<HloScatterInstruction>(instruction); in ImportInstructionImpl()