Sorry man, I don’t have much knowledge about tensorflow.
I found this link 152. See the last example of the page, it shows creation of a network. Also, I think as tensorflow supports complex dtypes, can you try changing the dtypes to your model inputs.