Searched defs:BorrowStream (Results 1 – 4 of 4) sorted by relevance
107 StatusOr<StreamPool::Ptr> Backend::BorrowStream(int device_ordinal) { in BorrowStream() function in xla::Backend112 StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) { in BorrowStream() function in xla::Backend
24 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
41 StreamPool::Ptr MlirGpuTestBase::BorrowStream() { in BorrowStream() function in xla::gpu::MlirGpuTestBase