Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_execution_profiler.cc54 bool do_profile, HloExecutionProfile* profile, se::Stream* stream, in HloExecutionProfiler() argument
56 : do_profile_(do_profile), in HloExecutionProfiler()
Dgpu_executable.cc141 bool do_profile = hlo_execution_profile != nullptr; in ExecuteThunks() local
142 if (do_profile) { in ExecuteThunks()
158 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream, in ExecuteThunks()
226 if (do_profile || block_host_until_done) { in ExecuteThunks()
247 if (do_profile) { in ExecuteThunks()
Dhlo_execution_profiler.h40 explicit HloExecutionProfiler(bool do_profile, HloExecutionProfile* profile,