Home
last modified time | relevance | path

Searched refs:hlo_profile_index_map (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.cc50 const HloProfileIndexMap& hlo_profile_index_map, in CreateHloProfilePrinterData() argument
56 size_t profile_counters_size = hlo_profile_index_map.total_count(); in CreateHloProfilePrinterData()
62 hlo_profile_index_map.computation_count()); in CreateHloProfilePrinterData()
65 hlo_profile_index_map.computation_to_profile_idx(); in CreateHloProfilePrinterData()
109 hlo_profile_index_map.GetProfileIndexFor(*hlo)); in CreateHloProfilePrinterData()
114 for (const auto& pair : hlo_profile_index_map.extra_metric_to_profile_idx()) { in CreateHloProfilePrinterData()
126 const HloProfileIndexMap* hlo_profile_index_map) in HloExecutionProfile() argument
128 hlo_profile_index_map_(*hlo_profile_index_map), in HloExecutionProfile()
Dexecutable.h66 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in Executable() argument
69 hlo_profile_index_map_(std::move(hlo_profile_index_map)) { in Executable()
153 const HloProfileIndexMap& hlo_profile_index_map() const { in hlo_profile_index_map() function
Dhlo_execution_profile.h99 const HloProfileIndexMap& hlo_profile_index_map,
111 const HloProfileIndexMap* hlo_profile_index_map);
Dexecutable.cc80 &hlo_profile_index_map()) in ExecuteOnStreamWrapper()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc493 std::unique_ptr<HloProfileIndexMap>* hlo_profile_index_map, in CreateHloProfilingArtifacts() argument
495 *hlo_profile_index_map = absl::make_unique<HloProfileIndexMap>(module); in CreateHloProfilingArtifacts()
502 (*hlo_profile_index_map)->instruction_to_profile_idx())); in CreateHloProfilingArtifacts()
515 **hlo_profile_index_map, cost_analysis, entry_computation.name()); in CreateHloProfilingArtifacts()
517 (*hlo_profile_index_map)->computation_to_profile_idx(); in CreateHloProfilingArtifacts()
620 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in RunBackend() local
625 &hlo_profile_index_map, &hlo_profile_printer_data)); in RunBackend()
717 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map))); in RunBackend()
845 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in CompileAheadOfTime() local
851 &hlo_profile_index_map, &hlo_profile_printer_data)); in CompileAheadOfTime()
Dcpu_executable.cc56 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable() argument
58 std::move(hlo_profile_index_map)), in CpuExecutable()
Dcpu_executable.h55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
Dgpu_executable.cc57 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in GpuExecutable() argument
59 std::move(hlo_profile_index_map)), in GpuExecutable()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dxla_hlo_profile_test.cc168 &executable->hlo_profile_index_map()); in ExecuteAndFetchProfile()