FinalExporter not working in TensorFlow 2.1 on Google AI-Platform

Question

I\'m trying to upgrade my model to use AI-Platform 2.1 instead of 1.15, but I can\'t get the FinalExporter to work.

I followed the steps outlined in ai-p