Home
last modified time | relevance | path

Searched refs:indexing_users_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dfusion_node_indexing_evaluation.cc31 indexing_users_[root].insert(fusion); in FusionNodeIndexingEvaluation()
104 for (const auto* user : indexing_users_.at(producer)) { in EvaluateEmittedInstructions()
113 CHECK(!indexing_users_.contains(producer)); in UpdateEvaluationCache()
114 indexing_users_[producer] = std::move(indexing_users_of_producer); in UpdateEvaluationCache()
123 std::move(indexing_users_.at(fusion_operand)); in RemoveFusionOperand()
124 indexing_users_.erase(fusion_operand); in RemoveFusionOperand()
145 for (const auto* user : indexing_users_[instruction]) { in UpdateIndexUsageCount()
170 indexing_users_[operand].insert(indexing_users_[instruction].begin(), in UpdateIndexingUsersOfOperands()
171 indexing_users_[instruction].end()); in UpdateIndexingUsersOfOperands()
175 indexing_users_[operand].insert(instruction); in UpdateIndexingUsersOfOperands()
Dfusion_node_indexing_evaluation.h83 indexing_users_; variable