Home
last modified time | relevance | path

Searched refs:tuple_sources (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc111 const PointsToSet::SourceSet& PointsToSet::tuple_sources( in tuple_sources() function in xla::PointsToSet
113 return tree_.element(index).tuple_sources; in tuple_sources()
118 tree_.mutable_element(index)->tuple_sources.insert(tuple); in add_tuple_source()
243 operand_points_to_set.tuple_sources(src_index)) { in HandleGetTupleElement()
310 for (auto& tuple_source : operand_points_to_set.tuple_sources(index)) { in HandleRecvDone()
348 operand_points_to_set.tuple_sources(src_index)) { in HandleSend()
384 operand_points_to_set.tuple_sources(src_index)) { in HandleTuple()
414 for (HloInstruction* tuple : false_points_to_set.tuple_sources(index)) { in HandleTupleSelect()
539 for (auto& tuple_source : src_points_to_set.tuple_sources(index)) { in CreateCopiedPointsToSet()
Dtuple_points_to_analysis_test.cc137 points_to_analysis_->GetPointsToSet(constant1).tuple_sources({}).empty()); in TEST_F()
144 points_to_analysis_->GetPointsToSet(constant2).tuple_sources({}).empty()); in TEST_F()
148 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({}), in TEST_F()
208 points_to_analysis_->GetPointsToSet(inner_tuple).tuple_sources({}), in TEST_F()
217 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({}), in TEST_F()
219 EXPECT_THAT(points_to_analysis_->GetPointsToSet(tuple).tuple_sources({0}), in TEST_F()
222 points_to_analysis_->GetPointsToSet(tuple).tuple_sources({1}).empty()); in TEST_F()
264 EXPECT_THAT(points_to_set.tuple_sources({}), in TEST_F()
526 EXPECT_THAT(points_to_set.tuple_sources({}), in TEST_F()
528 EXPECT_THAT(points_to_set.tuple_sources({0}), in TEST_F()
[all …]
Dtuple_points_to_analysis.h107 const SourceSet& tuple_sources(const ShapeIndex& index) const;
146 SourceSet tuple_sources; member