tensorflow code