Searched refs:profile_index_map (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile_test.cc | 54 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F() local
56 CreateHloProfilePrinterData(profile_index_map, cost_analysis, in TEST_F()
59 &profile_index_map); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 44 const HloProfileIndexMap* profile_index_map, in IrEmitterContext() argument
51 profile_index_map_(profile_index_map), in IrEmitterContext()
68 const HloProfileIndexMap* profile_index_map() { return profile_index_map_; } in profile_index_map() function
|
| D | gpu_compiler.cc | 554 int pointer_size, const HloProfileIndexMap* profile_index_map, in CompileModuleToLlvmIrImpl() argument
598 cuda_compute_capability, profile_index_map, &mlir_context, in CompileModuleToLlvmIrImpl()
880 std::unique_ptr<HloProfileIndexMap> profile_index_map; in RunBackend() local
892 profile_index_map = absl::make_unique<HloProfileIndexMap>(*module); in RunBackend()
894 CreateHloProfilePrinterData(*profile_index_map, cost_analysis, in RunBackend()
907 GetCanShareBuffer(), pointer_size_, profile_index_map.get(), &llvm_module, in RunBackend()
943 if (profile_index_map) { in RunBackend()
945 profile_index_map->GetProfileIndexFor(*module->entry_computation()); in RunBackend()
955 std::move(profile_index_map)}); in RunBackend()
|
| D | ir_emitter_unnested.cc | 3854 const auto* index_map = ir_emitter_context_->profile_index_map(); in BuildWhileThunk()
3882 const auto* index_map = ir_emitter_context_->profile_index_map(); in BuildForThunk()
3908 const auto* index_map = ir_emitter_context_->profile_index_map(); in BuildConditionalThunk()
5832 if (const auto* index_map = ir_emitter_context_->profile_index_map()) { in GetThunkInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 212 const std::unordered_map<const T*, int64>& profile_index_map);
|
| D | ir_emitter.cc | 2810 const std::unordered_map<const T*, int64>& profile_index_map) { in GetProfileCounterCommon() argument
2811 auto it = profile_index_map.find(&hlo); in GetProfileCounterCommon()
2812 if (it == profile_index_map.end()) { in GetProfileCounterCommon()
|