Home
last modified time | relevance | path

Searched refs:IsReductionFromOrToContiguousDimensions (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_fusible.cc44 !IsReductionFromOrToContiguousDimensions(instr)) { in IfFusedReadsElementsMultipleTimes()
83 if (IsReductionFromOrToContiguousDimensions(*operand)) { in IsReduceInputFusion()
92 IsReductionFromOrToContiguousDimensions( in IsReduceInputFusion()
104 IsReductionFromOrToContiguousDimensions(instr); in IsInputFusibleReduction()
119 if (IsReductionFromOrToContiguousDimensions(*inst)) { in GetRealHeroForMultiOutputFusion()
133 if (IsReductionFromOrToContiguousDimensions(*element_instr)) { in ShapesCompatibleForMultiOutputFusion()
145 if (IsReductionFromOrToContiguousDimensions(*instr_1) && in ShapesCompatibleForMultiOutputFusion()
146 IsReductionFromOrToContiguousDimensions(*instr_2) && in ShapesCompatibleForMultiOutputFusion()
191 !IsReductionFromOrToContiguousDimensions(instr) && in IsLoopFusible()
298 IsReductionFromOrToContiguousDimensions(instr)) { in SharedMemoryUsageNoCache()
[all …]
H A Dir_emission_utils.h64 bool IsReductionFromOrToContiguousDimensions(const HloInstruction& reduce);
67 bool IsReductionFromOrToContiguousDimensions(mlir::Operation* op);
H A Dir_emission_utils.cc244 bool IsReductionFromOrToContiguousDimensions(const HloInstruction& reduce) { in IsReductionFromOrToContiguousDimensions() function
250 bool IsReductionFromOrToContiguousDimensions(mlir::Operation* op) { in IsReductionFromOrToContiguousDimensions() function
460 if (IsReductionFromOrToContiguousDimensions(*inst)) { in IsFusedReductionOutputConsistent()
H A Dreduction_splitter.cc35 if (IsReductionFromOrToContiguousDimensions(*reduce)) { in HandleReduce()
H A Dtree_reduction_rewriter.cc60 if (!IsReductionFromOrToContiguousDimensions(*hlo)) { in HandleReduce()
H A Dmulti_output_fusion.cc65 (IsReductionFromOrToContiguousDimensions(*instr2) && in LegalToFuse()
H A Dir_emitter_unnested.cc1787 return IsReductionFromOrToContiguousDimensions(op); in HasAnyUnnestedReductionRoot()
4409 if (IsReductionFromOrToContiguousDimensions(fusion_roots[0])) { in IsUnrollingColumnReductionBeneficial()
4416 if (IsReductionFromOrToContiguousDimensions(op)) { in IsUnrollingColumnReductionBeneficial()
4699 if (IsReductionFromOrToContiguousDimensions(*hlo)) { in EmitIRForReduction()
4923 return IsReductionFromOrToContiguousDimensions(op); in EmitUnnestedReduction()
5002 if (IsReductionFromOrToContiguousDimensions(output_instruction)) { in EmitUnnestedReduction()
H A Dgpu_compiler.cc731 return IsReductionFromOrToContiguousDimensions(*r); in OptimizeHloPostLayoutAssignment()