Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dflags.cc53 .optimization_level_single_gpu = opt_level; in SetterForXlaAutoJitFlag()
61 .optimization_level_single_gpu = 1; in SetterForXlaAutoJitFlag()
75 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_single_gpu = in SetterForXlaAutoJitFlag()
190 mark_for_compilation_flags->xla_auto_jit_flag.optimization_level_single_gpu = in AllocateAndParseFlags()
H A Dflags.h40 int32 optimization_level_single_gpu; member
H A Dxla_cluster_util.cc242 if (flags->xla_auto_jit_flag.optimization_level_single_gpu != in GetXlaGlobalJitLevel()
245 flags->xla_auto_jit_flag.optimization_level_single_gpu); in GetXlaGlobalJitLevel()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc109 return static_cast<unsigned char>(flag.optimization_level_single_gpu > 0 || in TF_GetXlaAutoJitEnabled()