Searched refs:xla_global_jit_level (Results 1 – 2 of 2) sorted by relevance
291 XlaGlobalJitLevel xla_global_jit_level = in GetGlobalJitLevelForGraph() local293 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()300 : xla_global_jit_level.general; in GetGlobalJitLevelForGraph()
138 xla_config_registry::XlaGlobalJitLevel xla_global_jit_level = in IsXlaGlobalJitOn() local143 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()145 (xla_global_jit_level.general == OptimizerOptions::ON_1 || in IsXlaGlobalJitOn()146 xla_global_jit_level.general == OptimizerOptions::ON_2); in IsXlaGlobalJitOn()