search

tensorflow / custom-op

Guide for building custom op for TensorFlow
Apache License 2.0
378 stars 116 forks source link

Question about Building a Custom Op for Beam Search #99

Open StarWang opened 3 years ago

StarWang commented 3 years ago

Hi Team,

I'm trying to build a custom op for beam search and have a question on specifying the input type. My question is: can we take RNNCell as input? If so, can you share a reference or example on how to do this?

Thanks!