Searched refs:FusionNodeIndexingEvaluation (Results 1 – 8 of 8) sorted by relevance
27 FusionNodeIndexingEvaluation::FusionNodeIndexingEvaluation( in FusionNodeIndexingEvaluation() function in xla::FusionNodeIndexingEvaluation39 const int64 FusionNodeIndexingEvaluation::kAllowedCodeDuplication = 15;82 bool FusionNodeIndexingEvaluation::CodeDuplicationTooHigh( in CodeDuplicationTooHigh()90 bool FusionNodeIndexingEvaluation::MaxCodeDuplicationTooHigh() const { in MaxCodeDuplicationTooHigh()101 int64 FusionNodeIndexingEvaluation::EvaluateEmittedInstructions( in EvaluateEmittedInstructions()110 void FusionNodeIndexingEvaluation::UpdateEvaluationCache( in UpdateEvaluationCache()120 FusionNodeIndexingEvaluation::RemoveFusionOperand( in RemoveFusionOperand()129 void FusionNodeIndexingEvaluation::RecomputeCache() { in RecomputeCache()142 void FusionNodeIndexingEvaluation::UpdateIndexUsageCount( in UpdateIndexUsageCount()151 void FusionNodeIndexingEvaluation::UpdateIndexingUsersOfOperands( in UpdateIndexingUsersOfOperands()
25 class FusionNodeIndexingEvaluation {27 explicit FusionNodeIndexingEvaluation(const HloInstruction* fusion,
49 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()72 FusionNodeIndexingEvaluation(consumer)); in EvaluateEmittedInstructions()79 absl::flat_hash_map<const HloInstruction*, FusionNodeIndexingEvaluation>
113 FusionNodeIndexingEvaluation(consumer)); in ShouldFuse()139 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()
58 absl::flat_hash_map<const HloInstruction*, FusionNodeIndexingEvaluation>
149 FusionNodeIndexingEvaluation eval_consumer(consumer); in IsFusedIrEmitterInefficient()155 FusionNodeIndexingEvaluation eval_producer( in IsFusedIrEmitterInefficient()
133 FusionNodeIndexingEvaluation(consumer)); in ShouldFuse()211 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()
49 absl::flat_hash_map<const HloInstruction*, FusionNodeIndexingEvaluation>