Searched refs:HloProfileIndexMap (Results 1 – 10 of 10) sorted by relevance
35 class HloProfileIndexMap {38 explicit HloProfileIndexMap(const HloModule& module) in HloProfileIndexMap() function39 : HloProfileIndexMap(module, {}) {} in HloProfileIndexMap()40 explicit HloProfileIndexMap(const HloModule& module,43 HloProfileIndexMap(const HloProfileIndexMap&) = default;44 HloProfileIndexMap(HloProfileIndexMap&&) = default;46 HloProfileIndexMap& operator=(const HloProfileIndexMap&) = default;47 HloProfileIndexMap& operator=(HloProfileIndexMap&&) = default;99 const HloProfileIndexMap& hlo_profile_index_map,111 const HloProfileIndexMap* hlo_profile_index_map);[all …]
31 HloProfileIndexMap::HloProfileIndexMap(const HloModule& module, in HloProfileIndexMap() function in xla::HloProfileIndexMap50 const HloProfileIndexMap& hlo_profile_index_map, in CreateHloProfilePrinterData()126 const HloProfileIndexMap* hlo_profile_index_map) in HloExecutionProfile()
66 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in Executable()153 const HloProfileIndexMap& hlo_profile_index_map() const { in hlo_profile_index_map()207 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map_;
55 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F()
60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
774 std::unique_ptr<HloProfileIndexMap> profile_index_map; in RunBackend()786 profile_index_map = absl::make_unique<HloProfileIndexMap>(*module); in RunBackend()
57 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in GpuExecutable()
55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
493 std::unique_ptr<HloProfileIndexMap>* hlo_profile_index_map, in CreateHloProfilingArtifacts()495 *hlo_profile_index_map = absl::make_unique<HloProfileIndexMap>(module); in CreateHloProfilingArtifacts()620 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in RunBackend()845 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in CompileAheadOfTime()
56 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable()