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

Question

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

Answer 1

尝试添加此标志

tfcompile_flags = "--xla_cpu_multi_thread_eigen=false",

在tf_library规则内。