Searched refs:index_usage_count (Results 1 – 1 of 1) sorted by relevance
223 absl::flat_hash_map<const HloInstruction*, int64> index_usage_count; in IsFusedIrEmitterInefficient() local224 index_usage_count[consumer] = 1; in IsFusedIrEmitterInefficient()226 auto evaluate_fusion_computation = [&indexing_users, &index_usage_count]( in IsFusedIrEmitterInefficient()235 int64& total = index_usage_count[instruction]; in IsFusedIrEmitterInefficient()237 total = index_usage_count[fusion]; in IsFusedIrEmitterInefficient()250 total += index_usage_count[user] * weight; in IsFusedIrEmitterInefficient()278 int64& total = index_usage_count[producer]; in IsFusedIrEmitterInefficient()281 total += index_usage_count[user]; in IsFusedIrEmitterInefficient()294 for (const auto& entry : index_usage_count) { in IsFusedIrEmitterInefficient()300 return total > 8 * index_usage_count.size(); in IsFusedIrEmitterInefficient()