Searched refs:mlir_input (Results 1 – 2 of 2) sorted by relevance
167 Status EmitConstant(MlirEmitterInput mlir_input);191 Status EmitReduceFromMlir(MlirEmitterInput mlir_input);193 Status EmitSelectAndScatterFromMlir(MlirEmitterInput mlir_input);200 Status EmitRngGetAndUpdateState(MlirEmitterInput mlir_input);202 Status EmitScatterFromMlir(MlirEmitterInput mlir_input);204 Status EmitSortFromMlir(MlirEmitterInput mlir_input);206 Status EmitTriangularSolveFromMlir(MlirEmitterInput mlir_input);209 Status EmitNcclThunkFromMlir(MlirEmitterInput mlir_input);224 Status EmitOp(MlirEmitterInput mlir_input);293 Status EmitPadToStaticFromMlir(MlirEmitterInput mlir_input);[all …]
741 Status IrEmitterUnnested::EmitConstant(MlirEmitterInput mlir_input) { in EmitConstant() argument742 auto get_global = mlir::cast<mlir::GetGlobalMemrefOp>(mlir_input.op); in EmitConstant()814 Status IrEmitterUnnested::EmitPadToStaticFromMlir(MlirEmitterInput mlir_input) { in EmitPadToStaticFromMlir() argument816 auto pad_to_static = mlir::cast<mlir::lmhlo::CustomCallOp>(mlir_input.op); in EmitPadToStaticFromMlir()823 BuildKernelThunkForMlir(pad_to_static, mlir_input.thunk_info, in EmitPadToStaticFromMlir()824 mlir_input.extra_slice, &ir_arrays)); in EmitPadToStaticFromMlir()941 MlirEmitterInput mlir_input) { in EmitSliceToDynamicFromMlir() argument943 auto slice_to_dynamic = mlir::cast<mlir::lmhlo::CustomCallOp>(mlir_input.op); in EmitSliceToDynamicFromMlir()950 BuildKernelThunkForMlir(slice_to_dynamic, mlir_input.thunk_info, in EmitSliceToDynamicFromMlir()951 mlir_input.extra_slice, &ir_arrays)); in EmitSliceToDynamicFromMlir()[all …]