Home
last modified time | relevance | path

Searched refs:execution_profile (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dexecution_profile_test.cc40 ExecutionProfile execution_profile; in XLA_TEST_F() local
44 &execution_options_, &execution_profile)); in XLA_TEST_F()
47 << execution_profile.compute_cycle_count(); in XLA_TEST_F()
49 << execution_profile.compute_and_transfer_time_ns(); in XLA_TEST_F()
51 << execution_profile.compute_time_ns(); in XLA_TEST_F()
60 EXPECT_GT(execution_profile.compute_cycle_count(), 0); in XLA_TEST_F()
63 EXPECT_GT(execution_profile.compute_and_transfer_time_ns(), 0); in XLA_TEST_F()
64 EXPECT_GT(execution_profile.compute_time_ns(), 0); in XLA_TEST_F()
Dcompilation_cache_test.cc44 ExecutionProfile execution_profile; in ExecuteComputationR0F32() local
49 &execution_profile) in ExecuteComputationR0F32()
53 EXPECT_EQ(expect_cache_hit, execution_profile.compilation_cache_hit()); in ExecuteComputationR0F32()
61 ExecutionProfile execution_profile; in ExecuteComputationR2F32() local
64 &execution_options_, &execution_profile) in ExecuteComputationR2F32()
69 EXPECT_EQ(expect_cache_hit, execution_profile.compilation_cache_hit()); in ExecuteComputationR2F32()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h72 ExecutionProfile* execution_profile = nullptr);
93 ExecutionProfile* execution_profile = nullptr);
104 ExecutionProfile* execution_profile; member
109 ExecutionProfile* execution_profile) in XlaComputationInstance()
113 execution_profile(execution_profile) {} in XlaComputationInstance()
174 ExecutionProfile* execution_profile = nullptr);
Dclient.cc169 ExecutionProfile* execution_profile) { in ExecuteAndTransfer() argument
172 Execute(computation, arguments, execution_options, execution_profile)); in ExecuteAndTransfer()
253 ExecutionProfile* execution_profile) { in Execute() argument
270 if (execution_profile != nullptr) { in Execute()
271 *execution_profile = response.profile(); in Execute()
280 ExecutionProfile* execution_profile) { in Execute() argument
301 *execution_options, execution_profile}}; in Execute()
357 computations[i].execution_profile != nullptr) { in ExecuteParallel()
358 *computations[i].execution_profile = response.responses(i).profile(); in ExecuteParallel()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile_test.cc58 HloExecutionProfile execution_profile(profile_printer.get(), in TEST_F() local
64 execution_profile.SetCyclesTakenBy(add_instruction, add_cycles); in TEST_F()
65 execution_profile.SetCyclesTakenBy(dot_instruction, dot_cycles); in TEST_F()
71 EXPECT_THAT(execution_profile.ToString(clock_rate_ghz), in TEST_F()
Dexecutable.h311 ExecutionProfile* execution_profile, in PopulateExecutionProfile() argument
Dexecutable.cc195 state.profile = run_options->run_options().execution_profile(); in ExecuteWrapperBeforeExecution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc210 if (run_options->execution_profile()) { in ExecuteComputeFunction()
212 run_options->execution_profile()->set_compute_time_ns( in ExecuteComputeFunction()
216 run_options->execution_profile()->set_compute_cycle_count( in ExecuteComputeFunction()
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc91 ExecutionProfile* ExecutableRunOptions::execution_profile() const { in execution_profile() function in xla::ExecutableRunOptions
Dexecutable_run_options.h123 ExecutionProfile* execution_profile() const;
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc139 ExecutionProfile* profile = run_options->run_options().execution_profile(); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc241 if (run_options->run_options().execution_profile()) { in ExecuteThunks()
242 ExecutionProfile* profile = run_options->run_options().execution_profile(); in ExecuteThunks()