Searched refs:hlo_profiling_enabled (Results 1 – 12 of 12) sorted by relevance
53 bool hlo_profiling_enabled = in XLA_TEST_F() local59 if (hlo_profiling_enabled) { in XLA_TEST_F()
336 CHECK(hlo_profiling_enabled()); in hlo_profile_printer_data()341 CHECK(hlo_profiling_enabled()); in hlo_profile_index_map()348 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function
51 string key = absl::StrCat("profiling=", hlo_profiling_enabled()); in compilation_cache_key()
108 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function
206 executable.hlo_profiling_enabled() in ExecuteWrapperBeforeExecution()
393 executables[i]->hlo_profiling_enabled()) { in ExecuteParallelAndRegisterResult()
259 bool hlo_profiling_enabled() const { in hlo_profiling_enabled() function263 assert(hlo_profiling_enabled()); in hlo_profile_printer_data()
49 if (hlo_profiling_enabled()) { in XlaCompiledCpuFunction()
184 if (cpu_executable->hlo_profiling_enabled()) { in Compile()
701 if (module->config().hlo_profiling_enabled()) { in RunBackend()938 if (module->config().hlo_profiling_enabled()) { in CompileAheadOfTime()
883 if (module->config().hlo_profiling_enabled() || VLOG_IS_ON(1)) { in RunBackend()891 if (module->config().hlo_profiling_enabled()) { in RunBackend()
664 ASSERT_TRUE(fn.hlo_profiling_enabled()); in TEST()