Searched refs:GetReduceFromUnnestedMlir (Results 1 – 1 of 1) sorted by relevance
4200 static mlir::Operation* GetReduceFromUnnestedMlir(mlir::Operation* unnested_hlo, in GetReduceFromUnnestedMlir() function4225 GetReduceFromUnnestedMlir(unnested_hlo, index); in EmitPrologueForReduction()4420 GetReduceFromUnnestedMlir(unnested_hlo, index); in EmitEpilogueForReduction()4613 mlir::Operation* inst = GetReduceFromUnnestedMlir(unnested_hlo, index); in EmitTileElementForReduction()5403 auto reduce = GetReduceFromUnnestedMlir(unnested_hlo, index); in EmitIRForReduction()5547 GetReduceFromUnnestedMlir(unnested_hlo, i); in EmitReductionFromOrToContiguousDimensions()5572 GetReduceFromUnnestedMlir(unnested_hlo, i))) { in EmitReductionFromOrToContiguousDimensions()5573 first_reduce = GetReduceFromUnnestedMlir(unnested_hlo, i); in EmitReductionFromOrToContiguousDimensions()5581 GetReduceFromUnnestedMlir(unnested_hlo, i)); in EmitReductionFromOrToContiguousDimensions()