Searched defs:unnested_hlo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 383 const HloInstruction* unnested_hlo = hlo; in GetIndexTypeForKernel() local
4200 static mlir::Operation* GetReduceFromUnnestedMlir(mlir::Operation* unnested_hlo, in GetReduceFromUnnestedMlir()
4216 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitPrologueForReduction()
4378 llvm::Type* index_ty, mlir::Operation* unnested_hlo, in EmitEpilogueForReduction()
4595 mlir::Operation* unnested_hlo, const Shape& reduction_operand_shape, in EmitTileElementForReduction()
5233 bool IsUnrollingColumnReductionBeneficial(mlir::Operation* unnested_hlo, in IsUnrollingColumnReductionBeneficial()
5295 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce) { in ComputeReductionCodegenInfo()
5396 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitIRForReduction()
5532 mlir::Operation* unnested_hlo = mlir_input.op; in EmitReductionFromOrToContiguousDimensions() local
|
| D | ir_emission_utils.cc | 372 bool IsInputFusibleSlices(mlir::Operation* unnested_hlo, in IsInputFusibleSlices()
|