Searched refs:HloScatterInstruction (Results 1 – 6 of 6) sorted by relevance
2214 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2228 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2242 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2256 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2269 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2286 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2304 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2323 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2343 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()2358 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()[all …]
2394 HloScatterInstruction::HloScatterInstruction( in HloScatterInstruction() function in xla::HloScatterInstruction2408 string HloScatterInstruction::ScatterDimensionNumbersToString() const { in ScatterDimensionNumbersToString()2429 HloScatterInstruction::MakeScatterDimNumbers( in MakeScatterDimNumbers()2449 HloInstructionProto HloScatterInstruction::ToProto() const { in ToProto()2455 std::vector<string> HloScatterInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()2460 bool HloScatterInstruction::IdenticalSlowPath( in IdenticalSlowPath()2464 const auto& casted_other = static_cast<const HloScatterInstruction&>(other); in IdenticalSlowPath()2471 std::unique_ptr<HloInstruction> HloScatterInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()2475 return absl::make_unique<HloScatterInstruction>( in CloneWithNewOperandsImpl()
1399 class HloScatterInstruction : public HloInstruction {1401 explicit HloScatterInstruction(
1319 return absl::make_unique<HloScatterInstruction>( in CreateScatter()3526 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
1455 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()
1673 HloScatterInstruction::MakeScatterDimNumbers( in ParseInstructionRhs()