tensorflow 2.x cnn model