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 | 50 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()
|
D | executable.h | 66 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
|
D | hlo_execution_profile.h | 99 const HloProfileIndexMap& hlo_profile_index_map, 111 const HloProfileIndexMap* hlo_profile_index_map);
|
D | executable.cc | 80 &hlo_profile_index_map()) in ExecuteOnStreamWrapper()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_compiler.cc | 493 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()
|
D | cpu_executable.cc | 56 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable() argument 58 std::move(hlo_profile_index_map)), in CpuExecutable()
|
D | cpu_executable.h | 55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_executable.h | 60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
|
D | gpu_executable.cc | 57 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/ |
D | xla_hlo_profile_test.cc | 168 &executable->hlo_profile_index_map()); in ExecuteAndFetchProfile()
|