Searched refs:unnested_hlo (Results 1 – 4 of 4) sorted by relevance
72 bool IsInputFusibleSlices(mlir::Operation* unnested_hlo,
268 bool IsInputFusibleSlices(mlir::Operation* unnested_hlo, in IsInputFusibleSlices() argument270 auto fusion = mlir::dyn_cast<mlir::lmhlo::FusionOp>(unnested_hlo); in IsInputFusibleSlices()
328 const HloInstruction* unnested_hlo = hlo; in GetIndexTypeForKernel() local331 unnested_hlo = computation->FusionInstruction(); in GetIndexTypeForKernel()353 if (!shape_in_range(unnested_hlo->shape())) { in GetIndexTypeForKernel()362 if (!absl::c_all_of(unnested_hlo->operands(), hlo_shape_in_range)) { in GetIndexTypeForKernel()367 if (unnested_hlo->opcode() == HloOpcode::kFusion) { in GetIndexTypeForKernel()369 unnested_hlo->fused_instructions_computation()->instructions(), in GetIndexTypeForKernel()
131 bool IsSliceInputFusion(const HloInstruction& unnested_hlo) { in IsSliceInputFusion() argument132 if (!unnested_hlo.IsInputFusion()) { in IsSliceInputFusion()135 const HloInstruction* root = unnested_hlo.fused_expression_root(); in IsSliceInputFusion()