| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_value.h | 71 struct HloUse { struct
83 bool operator==(const HloUse& other) const { argument
89 bool operator!=(const HloUse& other) const { return !(*this == other); }
92 friend H AbslHashValue(H h, const HloUse& use) { in AbslHashValue() argument
98 std::ostream& operator<<(std::ostream& out, const HloUse& use);
151 absl::Span<const HloUse> GetUses() const { in GetUses()
153 [this](std::vector<HloUse>& uses) { ComputeUses(uses); }); in GetUses()
189 void ComputeUses(std::vector<HloUse>& uses) const;
197 Lazy<std::vector<HloUse>> uses_;
|
| D | hlo_value.cc | 59 std::string HloUse::ToString() const { in ToString()
67 std::ostream& operator<<(std::ostream& out, const HloUse& use) { in operator <<()
94 for (const HloUse& use : GetUses()) { in ToString()
156 void HloValue::ComputeUses(std::vector<HloUse>& uses) const { in ComputeUses()
177 HloUse new_use{user, i, position.index}; in ComputeUses()
180 for (const HloUse& use : uses) { in ComputeUses()
|
| D | hlo_dataflow_analysis_test.cc | 143 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
145 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
202 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
204 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
206 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
244 UnorderedElementsAre(HloUse{gte_out, 0, {0}})); in TEST_P()
249 UnorderedElementsAre(HloUse{gte_out, 0, {}})); in TEST_P()
251 UnorderedElementsAre(HloUse{gte_tuple, 0, {}})); in TEST_P()
305 UnorderedElementsAre(HloUse{call, 0, {}}, HloUse{add, 0, {}})); in TEST_P()
307 UnorderedElementsAre(HloUse{call, 1, {}}, HloUse{add, 1, {}})); in TEST_P()
[all …]
|
| D | memory_space_assignment.h | 142 float GetAlternateMemoryBenefit(const HloUse& use,
251 const HloUse* use) const = 0;
277 virtual void Begin(const HloUse& use, int64_t start_time,
347 const HloUse* use) const override;
359 void Begin(const HloUse& use, int64_t start_time, int64_t end_time) override;
412 const HloUse* use) const override;
424 void Begin(const HloUse& use, int64_t start_time, int64_t end_time) override;
494 std::function<bool(const HloUse&)>;
557 void AddUse(HloUse use);
589 const std::vector<HloUse>& uses() const { return uses_; } in uses()
[all …]
|
| D | hlo_ordering.cc | 214 absl::Span<const HloUse* const> uses, const HloValue& value, in UsesBeforeValueDefinition()
219 auto UseIsBeforeValueDefinition = [&](const HloUse& use) { in UsesBeforeValueDefinition()
419 std::vector<const HloUse*> uses; in LiveRangeStrictlyBefore()
420 for (const HloUse& use : a.GetUses()) { in LiveRangeStrictlyBefore()
|
| D | memory_space_assignment.cc | 115 absl::c_all_of(value.GetUses(), [&](const HloUse& use) { in IsCrossProgramPrefetchCandidate()
231 std::string UsesToString(const std::vector<HloUse>& uses) { in UsesToString()
305 for (const HloUse& use : value->GetUses()) { in GetMemoryBoundedness()
345 const HloUse& use, MemorySpaceAssignmentCostAnalysis::Cache* cache) const { in GetAlternateMemoryBenefit()
494 const HloUse* use) const { in LatestPrefetchStartTime()
517 void InstructionCountPrefetchIntervalPicker::Begin(const HloUse& use, in Begin()
681 const HloUse* use) const { in LatestPrefetchStartTime()
771 void CostAnalysisPrefetchIntervalPicker::Begin(const HloUse& use, in Begin()
1045 std::vector<HloUse> uses(value->GetUses().begin(), value->GetUses().end()); in CreateAllocationValues()
1046 absl::c_stable_sort(uses, [&](const HloUse& use1, const HloUse& use2) { in CreateAllocationValues()
[all …]
|
| D | hlo_alias_analysis.cc | 78 for (const HloUse& use : value.GetUses()) { in ComputeWhileAliasedValues()
185 for (const HloUse& use : value.GetUses()) { in ComputeInPlaceOperationAliasedValues()
205 for (const HloUse& use : value.GetUses()) { in ComputeAliasedValues()
|
| D | memory_space_propagation.cc | 106 for (const HloUse& use : value.GetUses()) { in Propagate()
|
| D | hlo_ordering.h | 82 absl::Span<const HloUse* const> uses, const HloValue& value,
|
| D | copy_insertion.cc | 1138 std::vector<const HloUse*> uses;
1220 for (const HloUse& use : value->GetUses()) { in AddValueList()
1289 for (const HloUse* use : p->uses) { in Verify()
1542 const HloUse* use) { in RemoveCopyValue()
1551 for (const HloUse* copy_use : copy_value_node->uses) { in RemoveCopyValue()
1737 [](std::string* s, const HloUse* use) { in ToString()
1903 for (const HloUse& use : value->GetUses()) { in AddSpecialCaseCopies()
|
| D | loop_schedule_linearizer.cc | 110 for (const HloUse& use : value_at_input.GetUses()) { in AddControlEdgesForLoopWrites()
|
| D | hlo_dataflow_analysis.cc | 1629 for (const HloUse& use : value->GetUses()) { in DoesNotUseOperandBuffer()
1879 for (const HloUse& use : in CanShareOperandBufferWithUser()
1881 if (use == HloUse{user, operand_and_output_index.first.operand_number, in CanShareOperandBufferWithUser()
1932 const HloUse& use = fusion_param_value.GetUses()[0]; in CanShareOperandBufferWithUser()
1988 absl::c_find_if(uses, [user](const HloUse& use) { in CanShareOperandBufferWithUser()
1993 absl::c_find_if(uses, [callee_root](const HloUse& use) { in CanShareOperandBufferWithUser()
|
| D | hlo_ordering_test.cc | 285 const HloUse* while_use = in TEST_F()
586 auto tuple_use = HloUse{root, 0}; in TEST_F()
|
| D | memory_space_assignment_test.cc | 4107 options.is_use_allowed_in_alternate_mem_fn = [](const HloUse& use) { in TEST_P()
4174 options.is_use_allowed_in_alternate_mem_fn = [](const HloUse& use) { in TEST_P()
6700 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
6707 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
6779 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
6786 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
6858 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
6865 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
6918 auto is_cross_program_prefetch = [](const HloUse& use) { in TEST_P()
6925 auto is_end_of_program_prefetch = [](const HloUse& use) { in TEST_P()
[all …]
|
| D | hlo_live_range.cc | 153 for (const HloUse& use : value.GetUses()) { in GetLastUsageTime()
|
| D | bfloat16_propagation.cc | 252 for (const HloUse& use : value->GetUses()) { in AllUsersConsumeBF16()
|
| D | buffer_assignment.cc | 887 for (const HloUse& use : buffer.GetUses()) { in BufferInfoString()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 499 bool NeedsInitValue(const HloUse& use) { in NeedsInitValue()
555 for (const HloUse& use : value.GetUses()) { in FindConstrainedUses()
|