| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_profile_printer.h | 23 #include "tensorflow/compiler/xla/service/hlo_profile_printer_data.pb.h"
27 // Pretty-print an array of profile counters using hlo_profile_printer_data.
28 string PrintHloProfile(const HloProfilePrinterData& hlo_profile_printer_data,
|
| hlo_profile_printer.cc | 23 string PrintHloProfile(const HloProfilePrinterData& hlo_profile_printer_data,
30 for (const auto& item : hlo_profile_printer_data.extra_metrics()) {
36 hlo_profile_printer_data.computation_infos()) {
51 hlo_profile_printer_data.entry_computation() == computation_info.name(),
|
| executable.h | 65 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
68 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)),
148 const HloProfilePrinterData& hlo_profile_printer_data() const { function in class:xla::Executable
|
| hlo_execution_profile.cc | 125 const HloProfilePrinterData* hlo_profile_printer_data,
127 : hlo_profile_printer_data_(*hlo_profile_printer_data),
|
| executable.cc | 79 ? absl::make_unique<HloExecutionProfile>(&hlo_profile_printer_data(),
|
| hlo_execution_profile.h | 110 HloExecutionProfile(const HloProfilePrinterData* hlo_profile_printer_data,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiled_cpu_function.h | 96 // hlo_profile_printer_data but xla::HloProfilePrinterData is forward
242 const xla::HloProfilePrinterData& hlo_profile_printer_data() const {
303 const xla::HloProfilePrinterData* hlo_profile_printer_data) {
304 static_data->hlo_profile_printer_data_ = hlo_profile_printer_data;
|
| xla_jit_compiled_cpu_function.cc | 163 &jit->static_data_, &cpu_executable->hlo_profile_printer_data());
166 cpu_executable->hlo_profile_printer_data().profile_counters_size());
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_compiler.h | 86 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data);
89 HloProfilePrinterData* hlo_profile_printer_data() const { function in class:xla::cpu::CpuAotCompilationResult
|
| cpu_executable.h | 54 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
|
| cpu_compiler.cc | 135 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data)
139 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)) {}
494 std::unique_ptr<HloProfilePrinterData>* hlo_profile_printer_data) {
514 *hlo_profile_printer_data = CreateHloProfilePrinterData(
621 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data;
625 &hlo_profile_index_map, &hlo_profile_printer_data));
717 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map)));
[all...] |
| cpu_executable.cc | 55 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
57 : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_executable.h | 59 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
|
| gpu_executable.cc | 56 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
58 : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data),
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| codegen.cc | 435 ? R"(#include "tensorflow/compiler/xla/service/hlo_profile_printer_data.pb.h")"
716 compile_result.aot->hlo_profile_printer_data()};
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| xla_hlo_profile_test.cc | 167 &executable->hlo_profile_printer_data(),
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| tfcompile_test.cc | 617 xla::PrintHloProfile(fn.hlo_profile_printer_data(), fn.profile_counters(),
|
