Searched refs:hlo_profiling_enabled (Results 1 – 13 of 13) sorted by relevance
149 CHECK(hlo_profiling_enabled()); in hlo_profile_printer_data()154 CHECK(hlo_profiling_enabled()); in hlo_profile_index_map()161 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function
42 string key = absl::StrCat("profiling=", hlo_profiling_enabled()); in compilation_cache_key()
72 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function
78 hlo_profiling_enabled() in ExecuteOnStreamWrapper()
443 executables[i]->hlo_profiling_enabled()) { in ExecuteParallelAndRegisterResult()
53 bool hlo_profiling_enabled = in XLA_TEST_F() local59 if (hlo_profiling_enabled) { in XLA_TEST_F()
239 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function243 assert(hlo_profiling_enabled()); in hlo_profile_printer_data()
46 if (hlo_profiling_enabled()) { in XlaCompiledCpuFunction()
161 if (cpu_executable->hlo_profiling_enabled()) { in Compile()
777 if (module->config().hlo_profiling_enabled() || VLOG_IS_ON(1)) { in RunBackend()785 if (module->config().hlo_profiling_enabled()) { in RunBackend()
281 if (hlo_profiling_enabled()) { in ExecuteAsyncOnStream()
622 if (module->config().hlo_profiling_enabled()) { in RunBackend()848 if (module->config().hlo_profiling_enabled()) { in CompileAheadOfTime()
598 ASSERT_TRUE(fn.hlo_profiling_enabled()); in TEST()