我正在测试一个将某些敏感计算卸载到安全环境中的库。Tensorflow 是我和我的团队有兴趣移植的一个应用程序，特别是与 XLA 一起工作。我的团队没有成功添加执行卸载的 TF 操作。
为了让它与 XLA 一起工作，我需要通过库提供的 API 插入向该库发送和接收数据的 XLA 操作。我的理解是这些XLA ops必须添加到tensorflow/compiler/tf2xla/下TF ops到XLA ops的翻译中。我猜 ComputationBuilder 定义中列出的操作是可用的 XLA 操作，调用外部函数目前不是其中之一。
是否可以添加这样一个可以降低到 llvm 调用指令的 XLA 操作？还是可以通过其他方式？
请分享你的想法。
谢谢 ！
PS：请注意，我使用 LLVM 的经验比使用 tensorflow 的经验要多。