Linearize seemingly doesn't work with `tf.data` api

I've been trying to use the linearize function, however it won't run the model - it just hangs after creating the contexts.

I've taken out the staging areas to ensure it isn't them, is linearize still being updated or has tensorflow now implemented your suggestions? 

System:
Ubuntu 16.04 
TF 1.4