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);
226 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in GetXlaGlobalJitLevel() argument229 if (jit_level_in_session_opts == OptimizerOptions::DEFAULT) { in GetXlaGlobalJitLevel()233 result.single_gpu = result.general = jit_level_in_session_opts; in GetXlaGlobalJitLevel()287 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts = in GetGlobalJitLevelForGraph() local292 GetXlaGlobalJitLevel(jit_level_in_session_opts); in GetGlobalJitLevelForGraph()
137 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in IsXlaGlobalJitOn() argument139 xla_config_registry::GetGlobalJitLevel(jit_level_in_session_opts); in IsXlaGlobalJitOn()153 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts) { in MemoryOptimizerEnabled() argument161 IsXlaGlobalJitOn(jit_level_in_session_opts)) { in MemoryOptimizerEnabled()