Home
last modified time | relevance | path

Searched refs:HloScatterInstruction (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc2916 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 …]
Dhlo_sharding_util.h180 const HloScatterInstruction& scatter);
Dhlo_instructions.cc2816 HloScatterInstruction::HloScatterInstruction( in HloScatterInstruction() function in xla::HloScatterInstruction
2833 /*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()
Dhlo_instructions.h1698 class HloScatterInstruction : public HloInstruction {
1700 explicit HloScatterInstruction(
Dhlo_sharding_util.cc1058 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
Dhlo_instruction.cc1630 return absl::make_unique<HloScatterInstruction>( in CreateScatter()
4246 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
Dhlo_instruction_test.cc1586 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()
Dalgebraic_simplifier.cc1014 !DynCast<HloScatterInstruction>(lhs)->indices_are_sorted() && in HandleAdd()
1015 !DynCast<HloScatterInstruction>(rhs)->indices_are_sorted(); in HandleAdd()
Dhlo_parser.cc2516 HloScatterInstruction::MakeScatterDimNumbers( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc594 Cast<const HloScatterInstruction>(use)->indices_are_sorted(); in CreateLiteralForConstrainedUses()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc508 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc545 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in GetScatterDimensionNumbers()
564 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in EmitScatterOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc431 auto scatter = Cast<HloScatterInstruction>(instruction); in ImportInstructionImpl()