Searched refs:HloExecutionProfile (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.cc | 124 HloExecutionProfile::HloExecutionProfile( in HloExecutionProfile() function in xla::HloExecutionProfile
133 void HloExecutionProfile::SetCyclesTakenBy(const HloInstruction* hlo, in SetCyclesTakenBy()
139 uint64 HloExecutionProfile::GetCyclesTakenBy(const HloInstruction& hlo) const { in GetCyclesTakenBy()
|
| D | executable.h | 85 HloExecutionProfile* hlo_execution_profile) = 0;
105 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
130 HloExecutionProfile* hlo_execution_profile, se::Stream* stream) { in PopulateExecutionProfile()
|
| D | dump.h | 33 class HloExecutionProfile; variable
60 const HloExecutionProfile& profile,
|
| D | hlo_execution_profile.h | 106 class HloExecutionProfile {
110 HloExecutionProfile(const HloProfilePrinterData* hlo_profile_printer_data,
|
| D | executable.cc | 76 std::unique_ptr<HloExecutionProfile> profile_ptr = in ExecuteOnStreamWrapper()
79 ? absl::make_unique<HloExecutionProfile>(&hlo_profile_printer_data(), in ExecuteOnStreamWrapper()
|
| D | hlo_graph_dumper.h | 63 const HloExecutionProfile* hlo_execution_profile = nullptr,
|
| D | hlo_execution_profile_test.cc | 59 HloExecutionProfile execution_profile(profile_printer.get(), in TEST_F()
|
| D | dump.cc | 186 const HloExecutionProfile* profile, string_view suffix, in DumpHloModuleImpl()
282 const HloExecutionProfile& profile, in DumpHloModuleIfEnabled()
|
| D | hlo_graph_dumper.cc | 311 const HloExecutionProfile* profile, NodeFilter filter) in HloDotDumper()
383 const HloExecutionProfile* profile_; // may be null
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 61 HloExecutionProfile* hlo_execution_profile) override;
97 HloExecutionProfile* hlo_execution_profile);
122 HloExecutionProfile* hlo_execution_profile);
|
| D | cpu_executable.cc | 139 HloExecutionProfile* hlo_execution_profile) { in ExecuteComputeFunction()
270 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
292 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStreamImpl()
330 HloExecutionProfile* hlo_execution_profile; in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_execution_profiler.h | 41 explicit HloExecutionProfiler(bool do_profile, HloExecutionProfile* profile,
73 HloExecutionProfile* profile_;
|
| D | gpu_executable.h | 82 HloExecutionProfile* hlo_execution_profile) override;
97 HloExecutionProfile* hlo_execution_profile);
|
| D | gpu_executable.cc | 69 HloExecutionProfile* hlo_execution_profile) { in ExecuteThunks()
239 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
|
| D | hlo_execution_profiler.cc | 55 bool do_profile, HloExecutionProfile* profile, se::Stream* stream, in HloExecutionProfiler()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.h | 52 HloExecutionProfile* hlo_execution_profile) override
|
| D | executable.cc | 51 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 166 HloExecutionProfile hlo_execution_profile( in ExecuteAndFetchProfile()
|