CODEWITHSUNDEEP
tensorflowtensorflow-xla
shamanqing
shamanqing

Reputation: 1

How to use tensorflow xla aot mode to build a CNN model?

I used xla aot build a CNN model, it compiled with no errors, but when execute the binary, a segmentation fault raised. and when i change the model to an fully connected network, it works fine. this seems a tensorflow bug, anyone have built the CNN model using xla aot?

Upvotes: 0

Views: 246

Answers (1)

Mohamed Ibrahim
Mohamed Ibrahim

Reputation: 1

Try to add this flag

tfcompile_flags = "--xla_cpu_multi_thread_eigen=false",

inside the tf_library rule.

Upvotes: 0

Related Questions