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)
61 StreamBorrower borrow_stream_;
Dbackend.h130 std::function<StatusOr<StreamPool::Ptr>(int)> StreamBorrower() { in StreamBorrower() function
Dservice.cc410 backend->StreamBorrower()); in ExecuteParallelAndRegisterResult()
502 run_options.emplace_back(options, backend->StreamBorrower()); in ExecuteAndRegisterResult()
Dhlo_runner.cc455 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc168 backend_->StreamBorrower()); in RunHelper()