Home
last modified time | relevance | path

Searched refs:StreamBorrower (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice_executable_run_options.h30 using StreamBorrower = std::function<StatusOr<StreamPool::Ptr>(int)>;
36 StreamBorrower borrow_stream = nullptr)
59 StreamBorrower borrow_stream_;
Dbackend.h130 std::function<StatusOr<StreamPool::Ptr>(int)> StreamBorrower() { in StreamBorrower() function
Dservice.cc455 backend->StreamBorrower()); in ExecuteParallelAndRegisterResult()
549 run_options.emplace_back(options, backend->StreamBorrower()); in ExecuteAndRegisterResult()
Dhlo_runner.cc453 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc169 backend_->StreamBorrower()); in Run()