我正在测试一个将某些敏感计算卸载到安全环境中的库。 Tensorflow是我和我的团队有兴趣移植的应用程序,尤其是与XLA一起工作。我的团队在添加TF卸载时没有成功。
为了使用XLA,我需要插入XLA操作,通过库提供的API向该库发送和接收数据。我的理解是这些XLA操作必须添加到tensorflow / compiler / tf2xla /下的TF ops到XLA ops的转换中。我猜测ComputationBuilder定义中列出的操作是可用的XLA操作,并且调用外部函数目前不是其中之一。
是否可以添加一个可以降低到llvm调用指令的XLA操作?或者通过其他方式可以吗?
请分享您的想法。
谢谢!
PS:请注意,我对LLVM的经验比使用tensorflow更多。