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