Lines Matching defs:backend
56 // Caches a XlaDeviceAllocator per <backend, device ordinal> pair. A
64 // backend and device_ordinal.
66 const xla::Backend* backend, int device_ordinal);
75 std::unordered_map<std::pair<const xla::Backend*, int>,
77 hash<std::pair<const xla::Backend*, int>>>
92 const xla::Backend* backend, int device_ordinal) {
96 auto it = state.allocators_.find({backend, device_ordinal});
102 xla::MakeUnique<XlaDeviceAllocator>(backend, device_ordinal);
104 state.allocators_[{backend, device_ordinal}] = std::move(alloc);
208 xla::Backend* backend = client()->mutable_backend();
210 backend, device_ordinal_);
217 xla::Backend* backend = client()->mutable_backend();
218 TF_ASSIGN_OR_RETURN(stream_, backend->BorrowStream(device_ordinal_));
