tensorflow 2.0 functional api