| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.cc | 125 HloExecutionProfile::HloExecutionProfile( in HloExecutionProfile() function in xla::HloExecutionProfile
134 void HloExecutionProfile::SetCyclesTakenBy(const HloInstruction* hlo, in SetCyclesTakenBy()
140 void HloExecutionProfile::SetCyclesTakenBy(size_t index, uint64 cycles_taken) { in SetCyclesTakenBy()
144 uint64 HloExecutionProfile::GetCyclesTakenBy(const HloInstruction& hlo) const { in GetCyclesTakenBy()
148 uint64 HloExecutionProfile::GetCyclesTakenBy(size_t index) const { in GetCyclesTakenBy()
152 HloExecutionProfileData HloExecutionProfile::ToProto() const { in ToProto()
|
| D | executable.cc | 84 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
106 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream()
121 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream()
187 std::shared_ptr<HloExecutionProfile> profile_ptr;
207 ? std::make_shared<HloExecutionProfile>( in ExecuteWrapperBeforeExecution()
269 std::shared_ptr<HloExecutionProfile> profile = state.profile_ptr; in ExecuteWrapperAfterExecution()
|
| D | executable.h | 261 HloExecutionProfile* hlo_execution_profile);
284 HloExecutionProfile* hlo_execution_profile);
291 HloExecutionProfile* hlo_execution_profile);
296 HloExecutionProfile* hlo_execution_profile) = 0;
312 HloExecutionProfile* hlo_execution_profile, se::Stream* stream) { in PopulateExecutionProfile()
|
| D | dump.h | 33 class HloExecutionProfile; variable
74 const HloExecutionProfile& profile,
|
| D | hlo_execution_profile.h | 106 class HloExecutionProfile {
108 HloExecutionProfile(const HloProfilePrinterData* hlo_profile_printer_data,
|
| D | hlo_graph_dumper.h | 74 const HloExecutionProfile* hlo_execution_profile = nullptr,
|
| D | hlo_execution_profile_test.cc | 58 HloExecutionProfile execution_profile(profile_printer.get(), in TEST_F()
|
| D | dump.cc | 221 const HloExecutionProfile* profile, in DumpHloModuleImpl()
427 const HloExecutionProfile& profile, in DumpHloModuleIfEnabled()
|
| D | hlo_graph_dumper.cc | 317 const HloExecutionProfile* profile, NodeFilter filter) in HloDotDumper()
389 const HloExecutionProfile* profile_; // may be null
1743 const HloExecutionProfile* hlo_execution_profile, in RenderGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 117 HloExecutionProfile* hlo_execution_profile) override;
122 HloExecutionProfile* hlo_execution_profile);
128 VariantArguments arguments, HloExecutionProfile* hlo_execution_profile);
143 HloExecutionProfile* hlo_execution_profile);
|
| D | hlo_execution_profiler.h | 40 explicit HloExecutionProfiler(bool do_profile, HloExecutionProfile* profile,
80 HloExecutionProfile* profile_;
|
| D | gpu_executable.cc | 131 HloExecutionProfile* hlo_execution_profile) { in ExecuteThunks()
401 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream()
409 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream()
418 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStreamImpl()
|
| D | hlo_execution_profiler.cc | 54 bool do_profile, HloExecutionProfile* profile, se::Stream* stream, in HloExecutionProfiler()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 61 HloExecutionProfile* hlo_execution_profile) override;
113 HloExecutionProfile* hlo_execution_profile);
|
| D | cpu_executable.cc | 151 HloExecutionProfile* hlo_execution_profile) { in ExecuteComputeFunction()
328 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream()
376 HloExecutionProfile* hlo_execution_profile; in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.h | 43 HloExecutionProfile* hlo_execution_profile) override;
|
| D | executable_base.cc | 44 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 50 HloExecutionProfile* hlo_execution_profile) override;
|
| D | tpu_executable_interface.cc | 171 HloExecutionProfile* /*hlo_execution_profile*/) { in ExecuteAsyncOnStream() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 166 HloExecutionProfile hlo_execution_profile( in ExecuteAndFetchProfile()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 102 HloExecutionProfile* hlo_execution_profile) override { in ExecuteAsyncOnStream()
|