MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / BorrowStream

Method BorrowStream

tensorflow/compiler/xla/service/backend.cc:108–111  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

106}
107
108StatusOr<StreamPool::Ptr> Backend::BorrowStream(int device_ordinal) {
109 TF_ASSIGN_OR_RETURN(auto executor, stream_executor(device_ordinal));
110 return BorrowStream(executor);
111}
112
113StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) {
114 tensorflow::mutex_lock l(mu_);

Callers 10

CopyDeviceTensorToCPUMethod · 0.45
EnsureStreamOkLockedMethod · 0.45
BorrowStreamForDeviceFunction · 0.45
TEST_FFunction · 0.45
ExecuteThunksMethod · 0.45
ExecuteLocallyMethod · 0.45
XLA_TEST_FFunction · 0.45
BM_LocalClientOverheadFunction · 0.45
TransferManagerTestMethod · 0.45
BM_DynamicSliceFunction · 0.45

Calls 3

containsMethod · 0.80
emplaceMethod · 0.45
atMethod · 0.45

Tested by 6

TEST_FFunction · 0.36
ExecuteLocallyMethod · 0.36
XLA_TEST_FFunction · 0.36
BM_LocalClientOverheadFunction · 0.36
TransferManagerTestMethod · 0.36
BM_DynamicSliceFunction · 0.36