Tf2 Python Api