Home
last modified time | relevance | path

Searched refs:jit_level_in_session_opts (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dxla_config_registry.cc44 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts) { in GetGlobalJitLevel() argument
48 return {jit_level_in_session_opts, jit_level_in_session_opts}; in GetGlobalJitLevel()
50 return state->getter(jit_level_in_session_opts); in GetGlobalJitLevel()
Dxla_config_registry.h45 OptimizerOptions::GlobalJitLevel jit_level_in_session_opts);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc229 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in GetXlaGlobalJitLevel() argument
232 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() local
295 GetXlaGlobalJitLevel(jit_level_in_session_opts); in GetGlobalJitLevelForGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc148 const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) { in IsXlaGlobalJitOn() argument
150 xla_config_registry::GetGlobalJitLevel(jit_level_in_session_opts); in IsXlaGlobalJitOn()