Add some new tensorflow ops

I think we could just send pr directly. Of course, we could make them be several prs, not one big pr.