Home
last modified time | relevance | path

Searched refs:AsyncExecution (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dexecution_tracker.cc27 AsyncExecution::AsyncExecution(Backend* backend, in AsyncExecution() function in xla::AsyncExecution
40 Status AsyncExecution::BlockUntilDone() const { in BlockUntilDone()
56 handle, absl::make_unique<AsyncExecution>(backend, std::move(streams), in Register()
76 StatusOr<const AsyncExecution*> ExecutionTracker::Resolve( in Resolve()
Dexecution_tracker.h41 class AsyncExecution {
43 AsyncExecution(Backend* backend, std::vector<StreamPool::Ptr> streams,
85 StatusOr<const AsyncExecution*> Resolve(const ExecutionHandle& handle);
93 std::map<int64, std::unique_ptr<AsyncExecution>> handle_to_execution_