TME in TensorFlow

To be supported in future. You are very welcome to contribute!