Searched refs:tuple_sources (Results 1 – 3 of 3) sorted by relevance
114 const PointsToSet::SourceSet& PointsToSet::tuple_sources( in tuple_sources() function in xla::PointsToSet116 return tree_.element(index).tuple_sources; in tuple_sources()121 tree_.mutable_element(index)->tuple_sources.insert(tuple); in add_tuple_source()246 operand_points_to_set.tuple_sources(src_index)) { in HandleGetTupleElement()313 for (auto& tuple_source : operand_points_to_set.tuple_sources(index)) { in HandleRecvDone()337 operand_points_to_set.tuple_sources(/*index=*/{})) { in HandleCopyStart()358 operand_points_to_set.tuple_sources(src_index)) { in HandleCopyDone()398 operand_points_to_set.tuple_sources(src_index)) { in HandleSend()434 operand_points_to_set.tuple_sources(src_index)) { in HandleTuple()464 for (HloInstruction* tuple : false_points_to_set.tuple_sources(index)) { in HandleTupleSelect()[all …]
139 points_to_analysis_->GetPointsToSet(constant1).tuple_sources({}).empty()); in TEST_F()146 points_to_analysis_->GetPointsToSet(constant2).tuple_sources({}).empty()); in TEST_F()150 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({}), in TEST_F()210 points_to_analysis_->GetPointsToSet(inner_tuple).tuple_sources({}), in TEST_F()219 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({}), in TEST_F()221 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({0}), in TEST_F()224 points_to_analysis_->GetPointsToSet(tuple).tuple_sources({1}).empty()); in TEST_F()266 EXPECT_THAT(points_to_set.tuple_sources({}), in TEST_F()554 EXPECT_THAT(points_to_set.tuple_sources({}), in TEST_F()556 EXPECT_THAT(points_to_set.tuple_sources({0}), in TEST_F()[all …]
108 const SourceSet& tuple_sources(const ShapeIndex& index) const;147 SourceSet tuple_sources; member