Searched refs:hlo_profile_index_map (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.cc | 51 const HloProfileIndexMap& hlo_profile_index_map, in CreateHloProfilePrinterData() argument
57 size_t profile_counters_size = hlo_profile_index_map.total_count(); in CreateHloProfilePrinterData()
63 hlo_profile_index_map.computation_count()); in CreateHloProfilePrinterData()
66 hlo_profile_index_map.computation_to_profile_idx(); in CreateHloProfilePrinterData()
110 hlo_profile_index_map.GetProfileIndexFor(*hlo)); in CreateHloProfilePrinterData()
115 for (const auto& pair : hlo_profile_index_map.extra_metric_to_profile_idx()) { in CreateHloProfilePrinterData()
127 const HloProfileIndexMap* hlo_profile_index_map) in HloExecutionProfile() argument
129 hlo_profile_index_map_(*hlo_profile_index_map), in HloExecutionProfile()
|
| D | hlo_execution_profile.h | 99 const HloProfileIndexMap& hlo_profile_index_map,
109 const HloProfileIndexMap* hlo_profile_index_map);
|
| D | executable.h | 242 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in Executable() argument
245 hlo_profile_index_map_(std::move(hlo_profile_index_map)) { in Executable()
340 const HloProfileIndexMap& hlo_profile_index_map() const { in hlo_profile_index_map() function
|
| D | executable.cc | 209 &executable.hlo_profile_index_map()) in ExecuteWrapperBeforeExecution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 543 std::unique_ptr<HloProfileIndexMap>* hlo_profile_index_map, in CreateHloProfilingArtifacts() argument
545 *hlo_profile_index_map = absl::make_unique<HloProfileIndexMap>(module); in CreateHloProfilingArtifacts()
552 (*hlo_profile_index_map)->instruction_to_profile_idx())); in CreateHloProfilingArtifacts()
565 **hlo_profile_index_map, cost_analysis, entry_computation.name()); in CreateHloProfilingArtifacts()
567 (*hlo_profile_index_map)->computation_to_profile_idx(); in CreateHloProfilingArtifacts()
699 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in RunBackend() local
704 &hlo_profile_index_map, &hlo_profile_printer_data)); in RunBackend()
792 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map))); in RunBackend()
935 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in CompileAheadOfTime() local
941 &hlo_profile_index_map, &hlo_profile_printer_data)); in CompileAheadOfTime()
|
| D | cpu_executable.h | 55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
|
| D | cpu_executable.cc | 60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable() argument
62 std::move(hlo_profile_index_map)), in CpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 84 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map = nullptr; member
|
| D | gpu_executable.cc | 60 std::move(params.hlo_profile_index_map)), in GpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 168 &executable->hlo_profile_index_map()); in ExecuteAndFetchProfile()
|