Searched defs:hlo_profile_index_map (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_execution_profile.cc | 44 const HloProfileIndexMap& hlo_profile_index_map, 49 size_t profile_counters_size = hlo_profile_index_map.total_count(); 55 hlo_profile_index_map.computation_count()); 58 hlo_profile_index_map.computation_to_profile_idx(); 101 hlo_profile_index_map.GetProfileIndexFor(*hlo)); 110 const HloProfileIndexMap* hlo_profile_index_map) 112 hlo_profile_index_map_(*hlo_profile_index_map), 43 CreateHloProfilePrinterData( const HloProfileIndexMap& hlo_profile_index_map, const HloCostAnalysis& cost_analysis) argument 108 HloExecutionProfile( const HloProfilePrinterData* hlo_profile_printer_data, const HloProfileIndexMap* hlo_profile_index_map) argument
|
H A D | executable.h | 50 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) 53 hlo_profile_index_map_(std::move(hlo_profile_index_map)) { 125 const HloProfileIndexMap& hlo_profile_index_map() const { function in class:xla::Executable 47 Executable( std::unique_ptr<const HloModule> hlo_module, std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) argument
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_executable.cc | 59 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) 61 std::move(hlo_profile_index_map)), 53 CpuExecutable( std::unique_ptr<SimpleOrcJIT> jit, std::unique_ptr<const BufferAssignment> assignment, std::unique_ptr<const HloModule> hlo_module, const string& entry_function_name, std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) argument
|
H A D | parallel_cpu_executable.cc | 65 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) 67 std::move(hlo_profile_index_map)), 57 ParallelCpuExecutable( std::unique_ptr<SimpleOrcJIT> jit, std::unique_ptr<const BufferAssignment> assignment, std::unique_ptr<const HloModule> hlo_module, std::unique_ptr<const HloInstructionMap<string>> function_names, std::unordered_map<const HloInstruction*, std::unique_ptr<unsigned char[]>> aligned_constants, std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) argument
|
H A D | cpu_compiler.cc | 489 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map; local 492 hlo_profile_index_map = MakeUnique<HloProfileIndexMap>(*module); 498 hlo_profile_index_map->instruction_to_profile_idx())); 511 CreateHloProfilePrinterData(*hlo_profile_index_map, cost_analysis); 513 hlo_profile_index_map->computation_to_profile_idx(); 624 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map))); 703 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map)));
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_executable.cc | 120 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) 122 std::move(hlo_profile_index_map)), 113 GpuExecutable( const string& ptx, const std::vector<uint8>& cubin, std::pair<int, int> compute_capability, std::unique_ptr<const ThunkSchedule> thunk_schedule, std::unique_ptr<const HloModule> hlo_module, std::unique_ptr<const BufferAssignment> assignment, std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) argument
|
Completed in 200 milliseconds