Home
last modified time | relevance | path

Searched refs:xla_auto_jit_flag (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dflags.cc48 mark_for_compilation_flags->xla_auto_jit_flag in SetterForXlaAutoJitFlag()
50 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_general = in SetterForXlaAutoJitFlag()
56 mark_for_compilation_flags->xla_auto_jit_flag in SetterForXlaAutoJitFlag()
58 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_general = in SetterForXlaAutoJitFlag()
71 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_single_gpu = in SetterForXlaAutoJitFlag()
145 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_single_gpu = in AllocateAndParseFlags()
147 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_general = 0; in AllocateAndParseFlags()
Dxla_cluster_util.cc239 if (flags->xla_auto_jit_flag.optimization_level_single_gpu != in GetXlaGlobalJitLevel()
242 flags->xla_auto_jit_flag.optimization_level_single_gpu); in GetXlaGlobalJitLevel()
244 if (flags->xla_auto_jit_flag.optimization_level_general != in GetXlaGlobalJitLevel()
247 flags->xla_auto_jit_flag.optimization_level_general); in GetXlaGlobalJitLevel()
Dflags.h50 XlaAutoJitFlag xla_auto_jit_flag; member