| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.h | 135 const HloScatterInstruction* scatter);
140 const HloScatterInstruction* scatter);
147 const HloScatterInstruction& scatter);
154 const HloScatterInstruction& scatter);
176 const HloSharding& update_sharding, const HloScatterInstruction& scatter);
182 const HloScatterInstruction& scatter);
193 const HloScatterInstruction& scatter);
|
| D | scatter_simplifier.cc | 96 HloScatterInstruction* scatter, in TransformScatterUpdates()
133 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction()
206 if (auto* scatter = DynCast<HloScatterInstruction>(inst)) { in InstructionMatchesPattern()
|
| D | shape_inference_test.cc | 3032 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3046 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3060 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3074 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3087 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3104 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3121 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3139 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3159 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
3174 HloScatterInstruction::MakeScatterDimNumbers( in TEST_P()
[all …]
|
| D | scatter_expander.cc | 249 HloScatterInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()
379 static int64_t ScatterTripCount(const HloScatterInstruction* scatter) { in ScatterTripCount()
414 auto* scatter = Cast<HloScatterInstruction>(inst); in ExpandInstruction()
488 auto* scatter = DynCast<HloScatterInstruction>(inst); in InstructionMatchesPattern()
|
| D | hlo_sharding_util.cc | 731 const HloScatterInstruction* scatter) { in ScatterIndexSharding()
774 const HloScatterInstruction* scatter) { in ScatterDataSharding()
816 const HloSharding& index_sharding, const HloScatterInstruction& scatter) { in ScatterEffectiveIndexSharding()
860 const HloScatterInstruction& scatter) { in ScatterEffectiveDataSharding()
1134 const HloSharding& update_sharding, const HloScatterInstruction& scatter) { in ScatterOutputShardingFromUpdate()
1153 const HloScatterInstruction& scatter) { in ScatterUpdateShardingFromOutput()
1173 const HloScatterInstruction& scatter) { in IdentityValueAndHloOpcodeForScatterReduceComputation()
|
| D | select_and_scatter_expander.cc | 183 HloScatterInstruction::MakeScatterDimNumbers( in ExpandInstruction()
|
| D | hlo_instructions.cc | 3140 HloScatterInstruction::HloScatterInstruction( in HloScatterInstruction() function in xla::HloScatterInstruction
3157 /*static*/ std::string HloScatterInstruction::ScatterDimensionNumbersToString( in ScatterDimensionNumbersToString()
3179 HloScatterInstruction::MakeScatterDimNumbers( in MakeScatterDimNumbers()
3199 HloInstructionProto HloScatterInstruction::ToProto() const { in ToProto()
3207 std::vector<std::string> HloScatterInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
3220 bool HloScatterInstruction::IdenticalSlowPath( in IdenticalSlowPath()
3224 const auto& casted_other = static_cast<const HloScatterInstruction&>(other); in IdenticalSlowPath()
3233 std::unique_ptr<HloInstruction> HloScatterInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()
3236 return std::make_unique<HloScatterInstruction>( in CloneWithNewOperandsImpl()
|
| D | hlo_dataflow_analysis.cc | 1773 const HloScatterInstruction* scatter = in GetInPlaceInputOutputPairs()
1774 DynCast<HloScatterInstruction>(instruction); in GetInPlaceInputOutputPairs()
|
| D | hlo_instructions.h | 2149 class HloScatterInstruction : public HloInstruction {
2151 explicit HloScatterInstruction(
|
| D | hlo_cost_analysis.cc | 1078 auto* scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter()
|
| D | sharding_propagation.cc | 1571 auto& scatter_user = *Cast<HloScatterInstruction>(&user); in GetShardingFromUser()
2182 auto* scatter = Cast<HloScatterInstruction>(instruction); in InferShardingFromOperands()
|
| D | hlo_instruction.cc | 1941 return absl::WrapUnique(new HloScatterInstruction( in CreateScatter()
1957 return std::make_unique<HloScatterInstruction>( in CreateScatter()
4828 return Cast<HloScatterInstruction>(this)->scatter_dimension_numbers(); in scatter_dimension_numbers()
|
| D | algebraic_simplifier.cc | 901 !DynCast<HloScatterInstruction>(lhs)->indices_are_sorted() && in HandleAdd()
902 !DynCast<HloScatterInstruction>(rhs)->indices_are_sorted(); in HandleAdd()
5964 auto* scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter()
|
| D | hlo_instruction_test.cc | 1685 HloScatterInstruction::MakeScatterDimNumbers( in TEST_F()
|
| D | hlo_evaluator.cc | 2840 auto* scatter = DynCast<HloScatterInstruction>(hlo); in HandleScatter()
|
| D | hlo_parser.cc | 2883 HloScatterInstruction::MakeScatterDimNumbers( in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 774 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterPassthroughOperand()
813 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterIndexPassthroughPartition()
952 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatterTrivialIndexedOperandDimension()
1008 const HloScatterInstruction* scatter, absl::Span<PartitionedHlo> operands, in PartitionScatter()
1044 auto scatter = Cast<HloScatterInstruction>(hlo); in HandleScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 653 Cast<const HloScatterInstruction>(use)->indices_are_sorted(); in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 616 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in GetScatterDimensionNumbers()
639 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in EmitScatterOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 796 auto scatter = Cast<HloScatterInstruction>(instruction); in ImportInstructionImpl()
|