Searched defs:BorrowStream (Results 1 – 3 of 3) sorted by relevance
108 StatusOr<StreamPool::Ptr> Backend::BorrowStream(int device_ordinal) { in BorrowStream() function in xla::Backend113 StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) { in BorrowStream() function in xla::Backend
23 StreamPool::Ptr StreamPool::BorrowStream(se::StreamExecutor* executor) { in BorrowStream() function in xla::StreamPool
53 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream() function