Searched refs:points_to_set (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | tuple_points_to_analysis.cc | 184 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 …]
|
D | tuple_points_to_analysis_test.cc | 76 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 …]
|
D | layout_assignment.cc | 133 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()
|
D | tuple_points_to_analysis.h | 311 std::unique_ptr<PointsToSet> points_to_set; member
|
D | buffer_assignment.cc | 377 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()
|