Home
last modified time | relevance | path

Searched refs:points_to_set (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc184 const PointsToSet& points_to_set = GetPointsToSet(instruction); variable
185 points_to_set.ForEachElement(
202 PointsToSet& points_to_set = CreateEmptyPointsToSet(hlo_instruction); in DefaultAction() local
203 points_to_set.ForEachMutableElement( in DefaultAction()
213 points_to_set.add_tuple_source({}, hlo_instruction); in DefaultAction()
225 PointsToSet& points_to_set = CreateEmptyPointsToSet(get_tuple_element); in HandleGetTupleElement() local
227 *PerInst(get_tuple_element->operand(0))->points_to_set; in HandleGetTupleElement()
231 points_to_set.ForEachMutableElement( in HandleGetTupleElement()
244 points_to_set.add_tuple_source(target_index, tuple); in HandleGetTupleElement()
255 PointsToSet& points_to_set = CreateCopiedPointsToSet(copy, copy->operand(0)); in HandleCopy() local
[all …]
Dtuple_points_to_analysis_test.cc76 void ExpectHasBuffers(const PointsToSet::BufferList& points_to_set, in ExpectHasBuffers() argument
79 EXPECT_THAT(points_to_set, UnorderedElementsAreArray(vec)); in ExpectHasBuffers()
85 const PointsToSet::BufferList& points_to_set, in ExpectHasTopLevelBuffers() argument
91 ExpectHasBuffers(points_to_set, buffers); in ExpectHasTopLevelBuffers()
96 const PointsToSet::BufferSet& points_to_set, in ExpectHasTopLevelBuffers() argument
99 PointsToSet::BufferList(points_to_set.begin(), points_to_set.end()), in ExpectHasTopLevelBuffers()
256 auto& points_to_set = points_to_analysis_->GetPointsToSet(get_tuple_element); in TEST_F() local
257 EXPECT_EQ(3, points_to_set.size()); in TEST_F()
258 EXPECT_FALSE(points_to_set.IsAmbiguous()); in TEST_F()
259 EXPECT_TRUE(points_to_set.IsDistinct()); in TEST_F()
[all …]
Dlayout_assignment.cc133 const auto& points_to_set = points_to_analysis_.GetPointsToSet(instruction); in GetBufferSet() local
134 points_to_set.ForEachElement( in GetBufferSet()
890 const PointsToSet& points_to_set = in CheckLayouts() local
892 TF_RETURN_IF_ERROR(points_to_set.ForEachElementWithStatus( in CheckLayouts()
1199 const PointsToSet& points_to_set = in PropagateUseConstraintToDefs() local
1201 return points_to_set.ForEachElementWithStatus( in PropagateUseConstraintToDefs()
Dtuple_points_to_analysis.h311 std::unique_ptr<PointsToSet> points_to_set; member
Dbuffer_assignment.cc377 const PointsToSet& points_to_set = points_to_analysis().GetPointsToSet(hlo); in GetInstructionAllocation() local
378 const LogicalBuffer* buffer = points_to_set.element(shape_index)[0]; in GetInstructionAllocation()