Add a transform to remove param and buffer shape check from prologue

Add a transform to remove param and buffer shape check from prologue #5337