KeyError: "The name 'input_word_ids:0' refers to a Tensor which does not exist. The operation, 'input_word_ids', does not exist in the graph."

[dshahrokhian]

While trying to convert a RoBERTa model, tf-trt complains about input_word_ids not being in the graph. Nonetheless, if I run:

saved_model_cli show --dir {saved_model_dir} --all

it clearly shows the input as part of the graph:

Defined Functions:
...
...
...
    Option #2
      Callable with:
        Argument #1
          input_word_ids: TensorSpec(shape=(None, 192), dtype=tf.int32, name='input_word_ids')
        Argument #2
          DType: bool
          Value: True
        Argument #3
          DType: NoneType
          Value: None
...
...
...

Any ideas?