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

Function BorrowStreamForDevice

tensorflow/compiler/xla/client/local_client.cc:35–41  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

33
34namespace {
35StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal,
36 Backend* backend) {
37 if (device_ordinal < 0) {
38 device_ordinal = backend->default_device_ordinal();
39 }
40 return backend->BorrowStream(device_ordinal);
41}
42} // namespace
43
44LocalExecutable::LocalExecutable(std::unique_ptr<Executable> executable,

Callers 1

RunHelperMethod · 0.85

Calls 2

BorrowStreamMethod · 0.45

Tested by

no test coverage detected