Searched refs:tf_xla_cpu_global_jit (Results 1 – 9 of 9) sorted by relevance
117 &mark_for_compilation_flags->tf_xla_cpu_global_jit, in AppendMarkForCompilationPassFlagsInternal()152 mark_for_compilation_flags->tf_xla_cpu_global_jit = false; in AllocateAndParseFlags()
29 GetMarkForCompilationPassFlags()->tf_xla_cpu_global_jit); in Run()
66 bool tf_xla_cpu_global_jit; member
194 tensorflow::GetMarkForCompilationPassFlags()->tf_xla_cpu_global_jit = true; in main()
72 // Whether --tf_xla_cpu_global_jit is enabled in TF_XLA_FLAGS.
1661 if (flags->tf_xla_cpu_global_jit) { in ShouldCompileClusterImpl()
77 flags->tf_xla_cpu_global_jit = true; in TF_EnableXLACompilation()95 bool original = flags->tf_xla_cpu_global_jit; in TF_SetTfXlaCpuGlobalJit()96 flags->tf_xla_cpu_global_jit = static_cast<bool>(enable); in TF_SetTfXlaCpuGlobalJit()134 flags->tf_xla_cpu_global_jit = true; in TF_CreateConfig()
112 enabled on CPU by additionally using the flag `--tf_xla_cpu_global_jit`:115 $ TF_XLA_FLAGS="--tf_xla_auto_jit=2 --tf_xla_cpu_global_jit" path/to/your/program
152 bool cpu_global_jit = flags->tf_xla_cpu_global_jit; in GetCompilationDevice()