Home
last modified time | relevance | path

Searched refs:execution_profile (Results 1 – 11 of 11) 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()
Dlocal_client.cc175 &service_options, run_options.execution_profile(), arguments); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile_test.cc59 HloExecutionProfile execution_profile(profile_printer.get(), in TEST_F() local
65 execution_profile.SetCyclesTakenBy(add_instruction, add_cycles); in TEST_F()
66 execution_profile.SetCyclesTakenBy(dot_instruction, dot_cycles); in TEST_F()
68 EXPECT_THAT(execution_profile.ToString( in TEST_F()
Dexecutable.cc109 profile->MergeFrom(execution_profile()); in ExecuteOnStreamWrapper()
Dexecutable.h143 ExecutionProfile execution_profile() const { in execution_profile() function
Dservice.cc495 profile->MergeFrom(executables[0]->execution_profile()); in ExecuteParallelAndRegisterResult()
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc75 ExecutionProfile* ExecutableRunOptions::execution_profile() const { in execution_profile() function in xla::ExecutableRunOptions
Dexecutable_run_options.h73 ExecutionProfile* execution_profile() const;