| /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 | hlo_execution_profile.cc | 126 const HloProfilePrinterData* hlo_profile_printer_data, in HloExecutionProfile() argument
128 : 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 | executable.h | 241 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, in Executable() argument
244 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)), in Executable()
335 const HloProfilePrinterData& hlo_profile_printer_data() const { in hlo_profile_printer_data() function
|
| D | hlo_execution_profile_data.proto | 20 import "tensorflow/compiler/xla/service/hlo_profile_printer_data.proto";
|
| D | hlo_execution_profile.h | 108 HloExecutionProfile(const HloProfilePrinterData* hlo_profile_printer_data,
|
| D | executable.cc | 208 &executable.hlo_profile_printer_data(), in ExecuteWrapperBeforeExecution()
|
| D | BUILD | 51 name = "hlo_profile_printer_data",
52 srcs = ["hlo_profile_printer_data.proto"],
60 protodeps = [":hlo_profile_printer_data"],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiled_cpu_function.h | 262 const xla::HloProfilePrinterData& hlo_profile_printer_data() const { in hlo_profile_printer_data() function
333 const xla::HloProfilePrinterData* hlo_profile_printer_data) { in set_static_data_hlo_profile_printer_data() argument
334 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 | 186 &jit->static_data_, &cpu_executable->hlo_profile_printer_data()); in Compile()
189 cpu_executable->hlo_profile_printer_data().profile_counters_size()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 173 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data) in CpuAotCompilationResult() argument
177 hlo_profile_printer_data_(std::move(hlo_profile_printer_data)) {} in CpuAotCompilationResult()
544 std::unique_ptr<HloProfilePrinterData>* hlo_profile_printer_data) { in CreateHloProfilingArtifacts() argument
564 *hlo_profile_printer_data = CreateHloProfilePrinterData( in CreateHloProfilingArtifacts()
700 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data; in RunBackend() local
704 &hlo_profile_index_map, &hlo_profile_printer_data)); in RunBackend()
792 std::move(hlo_profile_printer_data), std::move(hlo_profile_index_map))); in RunBackend()
936 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data; in CompileAheadOfTime() local
941 &hlo_profile_index_map, &hlo_profile_printer_data)); in CompileAheadOfTime()
1021 result_slice.index(), std::move(hlo_profile_printer_data))); in CompileAheadOfTime()
|
| D | cpu_compiler.h | 87 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data);
90 HloProfilePrinterData* hlo_profile_printer_data() const { in hlo_profile_printer_data() function
|
| D | cpu_executable.h | 54 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data,
|
| D | cpu_executable.cc | 59 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, in CpuExecutable() argument
61 : Executable(std::move(hlo_module), std::move(hlo_profile_printer_data), in CpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 83 std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data = nullptr; member
|
| D | gpu_executable.cc | 59 std::move(params.hlo_profile_printer_data), in GpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 467 ? R"(#include "tensorflow/compiler/xla/service/hlo_profile_printer_data.pb.h")" in GenerateHeader()
769 compile_result.aot->hlo_profile_printer_data()}; in GenerateMetadata()
|
| /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 | 683 xla::PrintHloProfile(fn.hlo_profile_printer_data(), fn.profile_counters(), in TEST()
|