Как определить функции в tf.estimator.export.ServingInputReceiver для двух входных слоев при экспорте модели keras

Question

Я хочу экспортировать модель tf.keras, используя export_savedmodel на основе tf.estimator.export.ServingInputReceiver или tf.estimator.export.TensorServingInputReceiver.Мои входные переменные:

  input_1 = tf.keras.layers.Input(tensor=f1)
  input_2 = tf.keras.layers.Input(tensor=f2)
  batch_size = input_1.get_shape()[0]
  ... # build layers
  model = tf.keras.models.Model([input_1, input_2], logits)

Функция обслуживания записывается в виде:

def serving_input_receiver_fn():
# Prep a placeholder where the input example will be fed in
    features = tf.placeholder(dtype=dtype, shape=[batch_size] + shape, name='input_tensor')
    return tf.estimator.export.ServingInputReceiver(features=features, receiver_tensors=features)

Код экспорта:

    eval_estimator.export_savedmodel(FLAGS.model_dir, serving_input_receiver_fn= 
    build_tensor_serving_input_receiver_fn(shape, dtype=tf.float32, batch_size=1))

Как определить формув коде экспорта, как в моей модели есть два входных слоя?