Searched refs:tuple_sources (Results 1 – 2 of 2) 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()247 operand_points_to_set.tuple_sources(src_index)) { in HandleGetTupleElement()314 for (auto& tuple_source : operand_points_to_set.tuple_sources(index)) { in HandleRecvDone()333 operand_points_to_set.tuple_sources(source_index)) { in HandleAsyncStart()355 for (HloInstruction* tuple : operand_points_to_set.tuple_sources(index)) { in HandleAsyncUpdate()377 operand_points_to_set.tuple_sources(src_index)) { in HandleAsyncDone()403 operand_points_to_set.tuple_sources(/*index=*/{})) { in HandleCopyStart()424 operand_points_to_set.tuple_sources(src_index)) { in HandleCopyDone()[all …]
106 const SourceSet& tuple_sources(const ShapeIndex& index) const;145 SourceSet tuple_sources; member