| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_profile_printer.cc | 23 string PrintHloProfile(const HloProfilePrinterData& hlo_profile_printer_data, in PrintHloProfile() argument
30 for (const auto& item : hlo_profile_printer_data.extra_metrics()) { in PrintHloProfile()
36 hlo_profile_printer_data.computation_infos()) { in PrintHloProfile()
51 hlo_profile_printer_data.entry_computation() == computation_info.name(), in PrintHloProfile()
|
| D | executable.h | 65 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, in Executable() argument
68 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)), in Executable()
148 const HloProfilePrinterData& hlo_profile_printer_data() const { in hlo_profile_printer_data() function
|
| D | hlo_execution_profile.cc | 125 const HloProfilePrinterData* hlo_profile_printer_data, in HloExecutionProfile() argument
127 : hlo_profile_printer_data_(*hlo_profile_printer_data), in HloExecutionProfile()
|
| D | hlo_profile_printer.h | 28 string PrintHloProfile(const HloProfilePrinterData& hlo_profile_printer_data,
|
| D | hlo_execution_profile.h | 110 HloExecutionProfile(const HloProfilePrinterData* hlo_profile_printer_data,
|
| D | executable.cc | 79 ? absl::make_unique<HloExecutionProfile>(&hlo_profile_printer_data(), in ExecuteOnStreamWrapper()
|
| D | BUILD | 38 name = "hlo_profile_printer_data",
39 srcs = ["hlo_profile_printer_data.proto"],
3510 ":hlo_profile_printer_data",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 135 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data) in CpuAotCompilationResult() argument
139 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)) {} in CpuAotCompilationResult()
494 std::unique_ptr<HloProfilePrinterData>* hlo_profile_printer_data) { in CreateHloProfilingArtifacts() argument
514 *hlo_profile_printer_data = CreateHloProfilePrinterData( in CreateHloProfilingArtifacts()
621 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data; in RunBackend() local
625 &hlo_profile_index_map, &hlo_profile_printer_data)); in RunBackend()
717 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map))); in RunBackend()
846 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data; in CompileAheadOfTime() local
851 &hlo_profile_index_map, &hlo_profile_printer_data)); in CompileAheadOfTime()
935 result_slice.index(), std::move(hlo_profile_printer_data))); in CompileAheadOfTime()
|
| D | cpu_compiler.h | 86 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data);
89 HloProfilePrinterData* hlo_profile_printer_data() const { in hlo_profile_printer_data() function
|
| D | cpu_executable.cc | 55 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, in CpuExecutable() argument
57 : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data), in CpuExecutable()
|
| D | cpu_executable.h | 54 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiled_cpu_function.h | 242 const xla::HloProfilePrinterData& hlo_profile_printer_data() const { in hlo_profile_printer_data() function
303 const xla::HloProfilePrinterData* hlo_profile_printer_data) { in set_static_data_hlo_profile_printer_data() argument
304 static_data->hlo_profile_printer_data_ = hlo_profile_printer_data; in set_static_data_hlo_profile_printer_data()
|
| D | xla_jit_compiled_cpu_function.cc | 163 &jit->static_data_, &cpu_executable->hlo_profile_printer_data()); in Compile()
166 cpu_executable->hlo_profile_printer_data().profile_counters_size()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 59 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
|
| D | gpu_executable.cc | 56 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, in GpuExecutable() argument
58 : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data), in GpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 435 ? R"(#include "tensorflow/compiler/xla/service/hlo_profile_printer_data.pb.h")" in GenerateHeader()
716 compile_result.aot->hlo_profile_printer_data()}; in GenerateMetadata()
|
| D | tfcompile.bzl | 269 "//tensorflow/compiler/xla/service:hlo_profile_printer_data",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 167 &executable->hlo_profile_printer_data(), in ExecuteAndFetchProfile()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 617 xla::PrintHloProfile(fn.hlo_profile_printer_data(), fn.profile_counters(), in TEST()
|