Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dfusion_node_indexing_evaluation.cc32 index_usage_count_[fusion] = root_usage_count; in FusionNodeIndexingEvaluation()
91 for (const auto& entry : index_usage_count_) { in MaxCodeDuplicationTooHigh()
105 total += index_usage_count_.at(user); in EvaluateEmittedInstructions()
125 CHECK(!index_usage_count_.contains(fusion_operand)); in RemoveFusionOperand()
146 total += index_usage_count_.at(user); in UpdateIndexUsageCount()
148 CHECK(index_usage_count_.emplace(instruction, total).second); in UpdateIndexUsageCount()
Dfusion_node_indexing_evaluation.h88 absl::flat_hash_map<const HloInstruction*, int64> index_usage_count_; variable