tensorflow - 如何使用tensorflow xla aot模式搭建CNN模型？

Question

我使用 xla aot 构建了一个 CNN 模型，它编译没有错误，但是在执行二进制文件时，出现了分段错误。当我将模型更改为完全连接的网络时，它工作正常。这似乎是一个 tensorflow 错误，有人使用 xla aot 构建了 CNN 模型吗？

Answer 1

尝试添加此标志

tfcompile_flags = "--xla_cpu_multi_thread_eigen=false",

规则内tf_library。