Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.h35 class HloProfileIndexMap {
38 explicit HloProfileIndexMap(const HloModule& module) in HloProfileIndexMap() function
39 : 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 …]
Dhlo_execution_profile.cc32 HloProfileIndexMap::HloProfileIndexMap(const HloModule& module, in HloProfileIndexMap() function in xla::HloProfileIndexMap
51 const HloProfileIndexMap& hlo_profile_index_map, in CreateHloProfilePrinterData()
127 const HloProfileIndexMap* hlo_profile_index_map) in HloExecutionProfile()
Dexecutable.h242 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_;
Dhlo_execution_profile_test.cc54 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h44 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_;
Dgpu_executable.h84 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map = nullptr;
Dgpu_compiler.cc554 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()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.h55 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map);
Dcpu_compiler.cc543 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()
Dcpu_executable.cc60 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) in CpuExecutable()