Searched refs:single_gpu (Results 1 – 3 of 3) sorted by relevance
231 result.single_gpu = result.general = OptimizerOptions::OFF; in GetXlaGlobalJitLevel()233 result.single_gpu = result.general = jit_level_in_session_opts; in GetXlaGlobalJitLevel()241 result.single_gpu = static_cast<OptimizerOptions::GlobalJitLevel>( in GetXlaGlobalJitLevel()293 if (xla_global_jit_level.single_gpu == xla_global_jit_level.general) { in GetGlobalJitLevelForGraph()295 << xla_global_jit_level.single_gpu; in GetGlobalJitLevelForGraph()296 return xla_global_jit_level.single_gpu; in GetGlobalJitLevelForGraph()299 IsSingleGpuGraph(**options.graph) ? xla_global_jit_level.single_gpu in GetGlobalJitLevelForGraph()
32 OptimizerOptions::GlobalJitLevel single_gpu; member
143 bool is_on = (xla_global_jit_level.single_gpu == OptimizerOptions::ON_1 || in IsXlaGlobalJitOn()144 xla_global_jit_level.single_gpu == OptimizerOptions::ON_2) && in IsXlaGlobalJitOn()