Searched refs:HloUse (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_value.h | 64 struct HloUse { struct
76 bool operator==(const HloUse& other) const { argument
82 bool operator!=(const HloUse& other) const { return !(*this == other); }
85 std::ostream& operator<<(std::ostream& out, const HloUse& use);
141 const std::vector<HloUse>& uses() const { return uses_; } in uses()
165 std::vector<HloUse> uses_;
|
| D | hlo_dataflow_analysis_test.cc | 133 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
135 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
192 UnorderedElementsAre(HloUse{add, 0, {}})); in TEST_P()
194 UnorderedElementsAre(HloUse{add, 1, {}})); in TEST_P()
196 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
234 UnorderedElementsAre(HloUse{gte_out, 0, {0}})); in TEST_P()
239 UnorderedElementsAre(HloUse{gte_out, 0, {}})); in TEST_P()
241 UnorderedElementsAre(HloUse{gte_tuple, 0, {}})); in TEST_P()
295 UnorderedElementsAre(HloUse{call, 0, {}}, HloUse{add, 0, {}})); in TEST_P()
297 UnorderedElementsAre(HloUse{call, 1, {}}, HloUse{add, 1, {}})); in TEST_P()
[all …]
|
| D | hlo_value.cc | 58 string HloUse::ToString() const { in ToString()
65 std::ostream& operator<<(std::ostream& out, const HloUse& use) { in operator <<()
104 for (const HloUse& use : uses()) { in ToString()
185 HloUse new_use{user, operand_number, position.index}; in SetPositionsAndComputeUses()
188 for (const HloUse& use : uses_) { in SetPositionsAndComputeUses()
|
| D | copy_insertion.cc | 467 std::vector<const HloUse*> uses;
538 for (const HloUse& use : value->uses()) { in AddValueList()
607 for (const HloUse* use : p->uses) { in Verify()
805 const HloUse* use) { in RemoveCopyValue()
814 for (const HloUse* copy_use : copy_value_node->uses) { in RemoveCopyValue()
837 for (const HloUse* use : a.uses) { in LiveRangeBefore()
919 [](string* s, const HloUse* use) { in ToString()
|
| D | hlo_dataflow_analysis.cc | 930 for (const HloUse& use : value->uses()) { in DoesNotUseOperandBuffer()
996 if (absl::c_count_if(fusion_param_value.uses(), [&](const HloUse& use) { in CanDoInPlaceDynamicUpdateSlice()
1004 for (const HloUse& use : fusion_param_value.uses()) { in CanDoInPlaceDynamicUpdateSlice()
1086 const HloUse& use = fusion_param_value.uses()[0]; in CanShareOperandBufferWithUser()
1129 absl::c_find_if(uses, [user](const HloUse& use) { in CanShareOperandBufferWithUser()
1134 absl::c_find_if(uses, [callee_root](const HloUse& use) { in CanShareOperandBufferWithUser()
|
| D | hlo_ordering.cc | 171 const HloUse& use, const HloValue& value, in UseIsBeforeValueDefinition()
271 for (const HloUse& use : a.uses()) { in LiveRangeStrictlyBefore()
|
| D | hlo_ordering.h | 54 bool UseIsBeforeValueDefinition(const HloUse& use, const HloValue& value,
|
| D | hlo_alias_analysis.cc | 214 for (const HloUse& use : value.uses()) { in ComputeWhileAliasedBuffers()
313 for (const HloUse& use : value.uses()) { in ComputeAliasedBuffers()
|
| D | hlo_ordering_test.cc | 279 const HloUse& while_use = dataflow->GetValueDefinedAt(xla_while).uses()[0]; in TEST_F()
|
| D | bfloat16_propagation.cc | 227 for (const HloUse& use : value->uses()) { in AllUsersConsumeBF16()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 370 bool NeedsInitValue(const HloUse& use) { in NeedsInitValue()
397 for (const HloUse& use : value.uses()) { in FindConstrainedUses()
|