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,109 const HloProfileIndexMap* hlo_profile_index_map);[all …]
32 HloProfileIndexMap::HloProfileIndexMap(const HloModule& module, in HloProfileIndexMap() function in xla::HloProfileIndexMap51 const HloProfileIndexMap& hlo_profile_index_map, in CreateHloProfilePrinterData()127 const HloProfileIndexMap* hlo_profile_index_map) in HloExecutionProfile()
242 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in Executable()340 const HloProfileIndexMap& hlo_profile_index_map() const { in hlo_profile_index_map()405 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map_;
54 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F()
44 const HloProfileIndexMap* profile_index_map, in IrEmitterContext()68 const HloProfileIndexMap* profile_index_map() { return profile_index_map_; } in profile_index_map()94 const HloProfileIndexMap* profile_index_map_;
84 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map = nullptr;
554 int pointer_size, const HloProfileIndexMap* profile_index_map, in CompileModuleToLlvmIrImpl()880 std::unique_ptr<HloProfileIndexMap> profile_index_map; in RunBackend()892 profile_index_map = absl::make_unique<HloProfileIndexMap>(*module); in RunBackend()
55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
543 std::unique_ptr<HloProfileIndexMap>* hlo_profile_index_map, in CreateHloProfilingArtifacts()545 *hlo_profile_index_map = absl::make_unique<HloProfileIndexMap>(module); in CreateHloProfilingArtifacts()699 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in RunBackend()935 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; in CompileAheadOfTime()
60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable()