Home
last modified time | relevance | path

Searched refs:hlo_execution_profile (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc139 HloExecutionProfile* hlo_execution_profile) { in ExecuteComputeFunction() argument
156 hlo_execution_profile ? hlo_execution_profile->profile_counters().size() in ExecuteComputeFunction()
159 hlo_execution_profile in ExecuteComputeFunction()
160 ? hlo_execution_profile->mutable_profile_counters()->data() in ExecuteComputeFunction()
198 if (hlo_execution_profile) { in ExecuteComputeFunction()
200 hlo_execution_profile->total_cycles_executed( in ExecuteComputeFunction()
270 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
273 ExecuteAsyncOnStreamImpl(run_options, arguments, hlo_execution_profile)); in ExecuteOnStream()
292 HloExecutionProfile* hlo_execution_profile) { in ExecuteAsyncOnStreamImpl() argument
330 HloExecutionProfile* hlo_execution_profile; in ExecuteAsyncOnStreamImpl() member
[all …]
Dcpu_executable.h61 HloExecutionProfile* hlo_execution_profile) override;
97 HloExecutionProfile* hlo_execution_profile);
122 HloExecutionProfile* hlo_execution_profile);
DBUILD243 "//tensorflow/compiler/xla/service:hlo_execution_profile",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc69 HloExecutionProfile* hlo_execution_profile) { in ExecuteThunks() argument
83 bool do_profile = hlo_execution_profile != nullptr; in ExecuteThunks()
97 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream, in ExecuteThunks()
179 hlo_execution_profile->total_cycles_executed( in ExecuteThunks()
239 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
286 hlo_execution_profile)); in ExecuteOnStream()
Dgpu_executable.h82 HloExecutionProfile* hlo_execution_profile) override;
97 HloExecutionProfile* hlo_execution_profile);
DBUILD283 "//tensorflow/compiler/xla/service:hlo_execution_profile",
394 "//tensorflow/compiler/xla/service:hlo_execution_profile",
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.h85 HloExecutionProfile* hlo_execution_profile) = 0;
105 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
130 HloExecutionProfile* hlo_execution_profile, se::Stream* stream) { in PopulateExecutionProfile() argument
Dhlo_graph_dumper.h63 const HloExecutionProfile* hlo_execution_profile = nullptr,
DBUILD731 ":hlo_execution_profile",
772 ":hlo_execution_profile",
906 ":hlo_execution_profile",
2217 name = "hlo_execution_profile",
2218 srcs = ["hlo_execution_profile.cc"],
2219 hdrs = ["hlo_execution_profile.h"],
2240 ":hlo_execution_profile",
3284 ":hlo_execution_profile",
/external/tensorflow/tensorflow/compiler/xla/tests/
Dxla_hlo_profile_test.cc166 HloExecutionProfile hlo_execution_profile( in ExecuteAndFetchProfile() local
180 executable->ExecuteOnStream(&run_options, args, &hlo_execution_profile)); in ExecuteAndFetchProfile()
185 hlo_execution_profile.ToString(executor->GetDeviceDescription()); in ExecuteAndFetchProfile()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.h52 HloExecutionProfile* hlo_execution_profile) override
Dexecutable.cc51 HloExecutionProfile* hlo_execution_profile) { in ExecuteOnStream() argument
DBUILD90 "//tensorflow/compiler/xla/service:hlo_execution_profile",