API in tensorFlow