Home
last modified time | relevance | path

Searched refs:HloUse (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_value.h70 struct HloUse { struct
82 bool operator==(const HloUse& other) const { argument
88 bool operator!=(const HloUse& other) const { return !(*this == other); }
91 friend H AbslHashValue(H h, const HloUse& use) { in AbslHashValue() argument
97 std::ostream& operator<<(std::ostream& out, const HloUse& use);
153 const std::vector<HloUse>& uses() const { return uses_; } in uses()
177 std::vector<HloUse> uses_;
Dmemory_space_assignment.h201 const HloUse* use) const = 0;
217 virtual void Begin(const HloUse& use, int64 start_time, int64 end_time) = 0;
277 const HloUse* use) const override;
284 void Begin(const HloUse& use, int64 start_time, int64 end_time) override;
325 const HloUse* use) const override;
332 void Begin(const HloUse& use, int64 start_time, int64 end_time) override;
398 std::function<bool(const HloUse&)>;
436 [](const HloUse&) { return true; };
536 void AddUse(HloUse use);
553 const std::vector<HloUse>& uses() const { return uses_; } in uses()
[all …]
Dhlo_value.cc59 string HloUse::ToString() const { in ToString()
66 std::ostream& operator<<(std::ostream& out, const HloUse& use) { in operator <<()
105 for (const HloUse& use : uses()) { in ToString()
190 HloUse new_use{user, i, position.index}; in SetPositionsAndComputeUses()
193 for (const HloUse& use : uses_) { in SetPositionsAndComputeUses()
Dhlo_dataflow_analysis_test.cc136 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
138 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
195 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
197 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
199 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
237 UnorderedElementsAre(HloUse{gte_out, 0, {0}})); in TEST_P()
242 UnorderedElementsAre(HloUse{gte_out, 0, {}})); in TEST_P()
244 UnorderedElementsAre(HloUse{gte_tuple, 0, {}})); in TEST_P()
298 UnorderedElementsAre(HloUse{call, 0, {}}, HloUse{add, 0, {}})); in TEST_P()
300 UnorderedElementsAre(HloUse{call, 1, {}}, HloUse{add, 1, {}})); in TEST_P()
[all …]
Dhlo_ordering.cc213 absl::Span<const HloUse* const> uses, const HloValue& value, in UsesBeforeValueDefinition()
217 auto UseIsBeforeValueDefinition = [&](const HloUse& use) { in UsesBeforeValueDefinition()
407 std::vector<const HloUse*> uses; in LiveRangeStrictlyBefore()
408 for (const HloUse& use : a.uses()) { in LiveRangeStrictlyBefore()
Dmemory_space_assignment.cc74 absl::c_all_of(value.uses(), [&](const HloUse& use) { in IsCrossProgramPrefetchCandidate()
200 for (const HloUse& use : value->uses()) { in GetMemoryBoundedness()
333 const HloUse* use) const { in LatestPrefetchStartTime()
344 void InstructionCountPrefetchIntervalPicker::Begin(const HloUse& use, in Begin()
474 const HloUse* use) const { in LatestPrefetchStartTime()
547 void CostAnalysisPrefetchIntervalPicker::Begin(const HloUse& use, in Begin()
778 std::vector<HloUse> uses(value->uses()); in CreateAllocationValues()
779 absl::c_stable_sort(uses, [&](const HloUse& use1, const HloUse& use2) { in CreateAllocationValues()
791 for (const HloUse& use : uses) { in CreateAllocationValues()
888 const AllocationValue& value, const HloUse& use) const { in IsUseAllowedInAlternateMemory()
[all …]
Dhlo_dataflow_analysis.cc1200 for (const HloUse& use : value->uses()) { in DoesNotUseOperandBuffer()
1221 /*static*/ std::vector<std::pair<HloUse, ShapeIndex>>
1224 return {{HloUse{instruction, 0, {}}, {}}}; in GetInPlaceInputOutputPairs()
1228 std::vector<std::pair<HloUse, ShapeIndex>> input_output_pairs; in GetInPlaceInputOutputPairs()
1252 HloUse{instruction, fusion_parameter->parameter_number(), in GetInPlaceInputOutputPairs()
1286 for (const HloUse& use : GetUniqueValueAt(operand, operand_index).uses()) { in CanShareOperandBufferWithUser()
1332 const HloUse& use = fusion_param_value.uses()[0]; in CanShareOperandBufferWithUser()
1377 absl::c_find_if(uses, [user](const HloUse& use) { in CanShareOperandBufferWithUser()
1382 absl::c_find_if(uses, [callee_root](const HloUse& use) { in CanShareOperandBufferWithUser()
Dcopy_insertion.cc479 std::vector<const HloUse*> uses;
556 for (const HloUse& use : value->uses()) { in AddValueList()
625 for (const HloUse* use : p->uses) { in Verify()
831 const HloUse* use) { in RemoveCopyValue()
840 for (const HloUse* copy_use : copy_value_node->uses) { in RemoveCopyValue()
937 [](string* s, const HloUse* use) { in ToString()
1028 const HloUse& operand = operand_and_output_index.first; in AddCopiesToResolveInterference()
Dhlo_alias_analysis.cc214 for (const HloUse& use : value.uses()) { in ComputeWhileAliasedBuffers()
320 const HloUse& operand = operand_and_output_index.first; in ComputeInPlaceOperationAliasedBuffers()
331 for (const HloUse& use : value.uses()) { in ComputeInPlaceOperationAliasedBuffers()
347 for (const HloUse& use : value.uses()) { in ComputeAliasedBuffers()
Dmemory_space_propagation.cc98 for (const HloUse& use : value.uses()) { in Propagate()
Dhlo_ordering.h74 bool UsesBeforeValueDefinition(absl::Span<const HloUse* const> uses,
Dloop_schedule_linearizer.cc110 for (const HloUse& use : value_at_input.uses()) { in AddControlEdgesForLoopWrites()
Dhlo_dataflow_analysis.h172 static std::vector<std::pair<HloUse, ShapeIndex>> GetInPlaceInputOutputPairs(
Dhlo_live_range.cc140 for (const HloUse& use : value->uses()) { in CalculateBufferStartEndMap()
Dbfloat16_propagation.cc256 for (const HloUse& use : value->uses()) { in AllUsersConsumeBF16()
614 const HloUse& operand = operand_and_output_index.first; in ResolveInconsistencyOfAliasingBuffersHelper()
Dmemory_space_assignment_test.cc4254 options.is_use_allowed_in_alternate_mem_fn = [](const HloUse& use) { in TEST_P()
4321 options.is_use_allowed_in_alternate_mem_fn = [](const HloUse& use) { in TEST_P()
5147 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
5154 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
5207 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
5214 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
5269 const HloUse use{root, /*operand_number=*/1, /*operand_index=*/{}}; in TEST_F()
5366 const HloUse use{root, /*operand_number=*/1, /*operand_index=*/{}}; in TEST_F()
5446 const HloUse use{root, /*operand_number=*/1, /*operand_index=*/{}}; in TEST_F()
5515 const HloUse use{conditional1, /*operand_number=*/1, /*operand_index=*/{0}}; in TEST_F()
Dhlo_ordering_test.cc285 const HloUse* while_use = &dataflow->GetValueDefinedAt(xla_while).uses()[0]; in TEST_F()
Dbuffer_assignment.cc840 for (const HloUse& use : buffer.uses()) { in BufferInfoString()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc479 bool NeedsInitValue(const HloUse& use) { in NeedsInitValue()
506 for (const HloUse& use : value.uses()) { in FindConstrainedUses()