| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 151 HloExecutionProfile* hlo_execution_profile) { in ExecuteComputeFunction() argument
172 hlo_execution_profile ? hlo_execution_profile->profile_counters().size() in ExecuteComputeFunction()
175 hlo_execution_profile in ExecuteComputeFunction()
176 ? hlo_execution_profile->mutable_profile_counters()->data() in ExecuteComputeFunction()
215 if (hlo_execution_profile) { in ExecuteComputeFunction()
217 hlo_execution_profile->total_cycles_executed( in ExecuteComputeFunction()
328 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream() argument
376 HloExecutionProfile* hlo_execution_profile; in ExecuteAsyncOnStream() member
382 &run_options.run_options(), *task_buffers, hlo_execution_profile)); in ExecuteAsyncOnStream()
389 hlo_execution_profile}); in ExecuteAsyncOnStream()
|
| D | cpu_executable.h | 61 HloExecutionProfile* hlo_execution_profile) override;
113 HloExecutionProfile* hlo_execution_profile);
|
| D | BUILD | 340 "//tensorflow/compiler/xla/service:hlo_execution_profile",
|
| /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 | gpu_executable.cc | 131 HloExecutionProfile* hlo_execution_profile) { in ExecuteThunks() argument
141 bool do_profile = hlo_execution_profile != nullptr; in ExecuteThunks()
158 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream, in ExecuteThunks()
248 profile->set_compute_cycle_count(hlo_execution_profile->GetCyclesTakenBy( in ExecuteThunks()
401 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream() argument
403 hlo_execution_profile); in ExecuteAsyncOnStream()
409 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream() argument
412 ExecuteAsyncOnStreamImpl(run_options, arguments, hlo_execution_profile)); in ExecuteAsyncOnStream()
418 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStreamImpl() argument
554 hlo_execution_profile)); in ExecuteAsyncOnStreamImpl()
|
| D | BUILD | 289 "//tensorflow/compiler/xla/service:hlo_execution_profile",
397 "//tensorflow/compiler/xla/service:hlo_execution_profile",
635 "//tensorflow/compiler/xla/service:hlo_execution_profile",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 84 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
86 ExecuteAsyncOnStream(run_options, arguments, hlo_execution_profile); in ExecuteOnStream()
106 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream() argument
114 hlo_execution_profile)); in ExecuteAsyncOnStream()
121 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
123 run_options, std::move(arguments), hlo_execution_profile); in ExecuteOnStream()
|
| 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() argument
|
| D | hlo_graph_dumper.h | 74 const HloExecutionProfile* hlo_execution_profile = nullptr,
|
| D | hlo_graph_dumper.cc | 1743 const HloExecutionProfile* hlo_execution_profile, in RenderGraph() argument
1752 hlo_execution_profile, NodeFilter()) in RenderGraph()
|
| D | BUILD | 950 ":hlo_execution_profile",
992 ":hlo_execution_profile",
1133 ":hlo_execution_profile",
3046 name = "hlo_execution_profile",
3047 srcs = ["hlo_execution_profile.cc"],
3048 hdrs = ["hlo_execution_profile.h"],
3069 ":hlo_execution_profile",
4393 ":hlo_execution_profile",
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.h | 43 HloExecutionProfile* hlo_execution_profile) override;
|
| D | BUILD | 90 "//tensorflow/compiler/xla/service:hlo_execution_profile",
118 "//tensorflow/compiler/xla/service:hlo_execution_profile",
|
| D | executable_base.cc | 44 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStream() argument
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 50 HloExecutionProfile* hlo_execution_profile) override;
|
| D | BUILD | 351 "//tensorflow/compiler/xla/service:hlo_execution_profile",
|
| D | tpu_executor_c_api.h | 310 SE_HloExecutionProfile* hlo_execution_profile,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 166 HloExecutionProfile hlo_execution_profile( in ExecuteAndFetchProfile() local
180 executable->ExecuteOnStream(&run_options, args, &hlo_execution_profile)); in ExecuteAndFetchProfile()
184 *profile_output = hlo_execution_profile.ToString( in ExecuteAndFetchProfile()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 102 HloExecutionProfile* hlo_execution_profile) override { in ExecuteAsyncOnStream() argument
|