Searched refs:jit_level_in_session_opts (Results 1 – 4 of 4) sorted by relevance
44 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts) { in GetGlobalJitLevel() argument48 return {jit_level_in_session_opts, jit_level_in_session_opts}; in GetGlobalJitLevel()50 return state->getter(jit_level_in_session_opts); in GetGlobalJitLevel()
45 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts);
229 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in GetXlaGlobalJitLevel() argument232 if (jit_level_in_session_opts == OptimizerOptions::DEFAULT) { in GetXlaGlobalJitLevel()236 result.single_gpu = result.general = jit_level_in_session_opts; in GetXlaGlobalJitLevel()290 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts = in GetGlobalJitLevelForGraph() local295 GetXlaGlobalJitLevel(jit_level_in_session_opts); in GetGlobalJitLevelForGraph()
148 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in IsXlaGlobalJitOn() argument150 xla_config_registry::GetGlobalJitLevel(jit_level_in_session_opts); in IsXlaGlobalJitOn()