Home
last modified time | relevance | path

Searched refs:ExecutionProfile (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.h120 ExecutionProfile* profile = nullptr);
125 ExecutionProfile* profile = nullptr);
129 ExecutionProfile* profile = nullptr);
133 ExecutionProfile* profile = nullptr);
140 bool run_hlo_passes = true, ExecutionProfile* profile = nullptr);
145 bool run_hlo_passes = true, ExecutionProfile* profile = nullptr);
153 ExecutionProfile* profile = nullptr);
158 ExecutionProfile* profile = nullptr);
Dexecution_tracker.h44 const ExecutionProfile& profile, GlobalDataHandle result);
50 const ExecutionProfile& profile() const { return profile_; } in profile()
60 ExecutionProfile profile_;
76 const ExecutionProfile& profile,
Dhlo_runner.cc143 ExecutionProfile* profile) { in Execute()
158 ExecutionProfile* profile) { in Execute()
175 ExecutionProfile* profile) { in Execute()
188 ExecutionProfile* profile) { in Execute()
204 ExecutionProfile* profile) { in ExecuteWithDeviceBuffers()
225 ExecutionProfile* profile) { in ExecuteWithDeviceBuffers()
241 ExecutionProfile* profile) { in ExecuteWithDeviceBuffers()
260 ExecutionProfile* profile) { in ExecuteWithDeviceBuffers()
Dexecutable.h138 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile,
143 ExecutionProfile execution_profile() const { in execution_profile()
192 ExecutionProfile execution_profile_ GUARDED_BY(mutex_);
Dexecution_tracker.cc29 const ExecutionProfile& profile, in AsyncExecution()
51 const ExecutionProfile& profile, in Register()
Dservice.h255 const string& result_tag, ExecutionProfile* profile);
264 absl::Span<const string> result_tags, ExecutionProfile* profile);
Dexecutable.cc64 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile, in ExecuteOnStreamWrapper()
Dservice.cc398 absl::Span<const string> result_tags, ExecutionProfile* profile) { in ExecuteParallelAndRegisterResult()
521 const string& result_tag, ExecutionProfile* profile) { in ExecuteAndRegisterResult()
717 ExecutionProfile profile; in ExecuteGraphParallel()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h72 ExecutionProfile* execution_profile = nullptr);
93 ExecutionProfile* execution_profile = nullptr);
104 ExecutionProfile* execution_profile;
109 ExecutionProfile* execution_profile) in XlaComputationInstance()
174 ExecutionProfile* execution_profile = nullptr);
233 const ExecutionProfile& profile);
Dclient.cc169 ExecutionProfile* execution_profile) { in ExecuteAndTransfer()
253 ExecutionProfile* execution_profile) { in Execute()
280 ExecutionProfile* execution_profile) { in Execute()
467 const XlaComputation& computation, const ExecutionProfile& profile) { in ExecutionStatsAsString()
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.h36 class ExecutionProfile; variable
73 ExecutionProfile* execution_profile() const;
74 ExecutableRunOptions& set_execution_profile(ExecutionProfile* profile);
89 ExecutionProfile* execution_profile_ = nullptr;
Dexecutable_run_options.cc70 ExecutionProfile* profile) { in set_execution_profile()
75 ExecutionProfile* ExecutableRunOptions::execution_profile() const { in execution_profile()
Dxla.proto429 ExecutionProfile profile = 2;
442 ExecutionProfile profile = 2;
Dxla_data.proto265 message ExecutionProfile { message
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcompilation_cache_test.cc44 ExecutionProfile execution_profile; in ExecuteComputationR0F32()
61 ExecutionProfile execution_profile; in ExecuteComputationR2F32()
Dhlo_test_base.h241 ExecutionProfile* profile = nullptr,
245 std::vector<ExecutionProfile>* profiles,
Dexecution_profile_test.cc40 ExecutionProfile execution_profile; in XLA_TEST_F()
Dhlo_test_base.cc344 ExecutionProfile* profile, in Run()
394 std::vector<ExecutionProfile>* profiles, string backend_config) { in RunMultipleTimes()
Dlocal_client_execute_test.cc113 ExecutionProfile profile; in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc60 xla::ExecutionProfile profile; in main()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc285 ExecutionProfile profile; in ReplayComputation()