Tensorflow - XLA | 在运行时传递张量外部函数(Tensorflow - X

2019-09-28 12:54发布

我测试的是卸载某些敏感计算到一个安全的环境的库。 Tensorflow是一个应用程序,我和我的团队有兴趣端口这个工作,特别是与XLA工作。 我的团队没有成功的加入了TF的运算,做了卸载。

对于这个与XLA工作,我需要插入,通过库提供的API发送和接收数据到该库XLA欢声笑语。 我的理解是,这些XLA OPS必须添加到TF OPS下tensorflow /编译器/ tf2xla /翻译,以XLA欢声笑语。 我猜测,在ComputationBuilder的定义中列出的操作是可用的XLA OPS和调用外部函数不是目前他们中的一个。

是否有可能加入这样的XLA OP可以降低到LLVM调用指令? 或者是有可能通过其他方式?

请分享你的想法。

谢谢 !

PS:请注意,我有LLVM比tensorflow一个更有经验。

文章来源: Tensorflow - XLA | Passing tensors to external functions at runtime