Get wav2vec tensors

I want to make a serving signature for wav2vec in tensorflow and I don’t know how to get the input and output tensors .
this is the kind of signature I am trying to make :