Searched refs:IsReductionToVector (Results 1 – 6 of 6) sorted by relevance
67 if (IsReductionToVector(*operand)) { in IsReduceInputFusion()76 IsReductionToVector(*instr.fused_expression_root())) { in IsReduceInputFusion()86 return IsReduceInputFusion(instr) || IsReductionToVector(instr); in IsInputFusibleReduction()101 if (IsReductionToVector(*inst)) { in ShapesCompatibleForMultiOutputFusion()117 if (IsReductionToVector(*element_instr)) { in ShapesCompatibleForMultiOutputFusion()130 if (IsReductionToVector(*instr_1) && IsReductionToVector(*instr_2) && in ShapesCompatibleForMultiOutputFusion()175 !IsReductionToVector(instr)) || in IsLoopFusible()
205 if (IsReductionToVector(*producer)) { in ShouldFuseInexpensiveChecks()281 if (IsReductionToVector(*consumer) || in ChooseKind()
151 bool IsReductionToVector(const HloInstruction& reduce);
95 (IsReductionToVector(*instr2) && in LegalToFuse()
160 bool IsReductionToVector(const HloInstruction& reduce) { in IsReductionToVector() function
727 if (IsReductionToVector(*reduce)) { in HandleReduce()2651 CHECK(IsReductionToVector(*reduce_inst)); in EmitPrologueForOneReduction()