ML-driven Hardware Cost Model for MLIR