I am using TFServing to allow prediction on my model trained with TF2. After the training I save my model with a custom signature. Let\'s say my input is a tensor with shape