Searched refs:fused_reduce (Results 1 – 2 of 2) sorted by relevance
668 auto fused_reduce = b.create<mlir::mhlo::ReduceOp>( in EmitUsingElementalIrEmitter() local670 fused_reduce.body().takeBody(reduce.body()); in EmitUsingElementalIrEmitter()671 CHECK_EQ(fused_reduce.getNumResults(), reduce.out().size()); in EmitUsingElementalIrEmitter()674 b.create<mlir::TensorStoreOp>(loc, fused_reduce.getResult(i), in EmitUsingElementalIrEmitter()678 HloFunctionImporter::SetLayoutForMlir(fused_reduce, shape); in EmitUsingElementalIrEmitter()
832 fused_reduce {846 kind=kInput, calls=fused_reduce