Searched defs:hlo_profile_index_map (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_execution_profile.cc44 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 Dexecutable.h50 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 Dcpu_executable.cc59 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 Dparallel_cpu_executable.cc65 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 Dcpu_compiler.cc489 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 Dgpu_executable.cc120 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