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

时间:2017-07-13 07:51:07

标签: security machine-learning tensorflow tensorflow-xla

我正在测试一个将某些敏感计算卸载到安全环境中的库。 Tensorflow是我和我的团队有兴趣移植的应用程序,尤其是与XLA一起工作。我的团队在添加TF卸载时没有成功。

为了使用XLA,我需要插入XLA操作,通过库提供的API向该库发送和接收数据。我的理解是这些XLA操作必须添加到tensorflow / compiler / tf2xla /下的TF ops到XLA ops的转换中。我猜测ComputationBuilder定义中列出的操作是可用的XLA操作,并且调用外部函数目前不是其中之一。

是否可以添加一个可以降低到llvm调用指令的XLA操作?或者通过其他方式可以吗?

请分享您的想法。

谢谢!

PS:请注意,我对LLVM的经验比使用tensorflow更多。

0 个答案:

没有答案