- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for UsePjRtForSingleDeviceCompilation (0.28 sec)
-
tensorflow/compiler/jit/xla_device.cc
} Allocator* XlaDevice::GetAllocatorLocked(AllocatorAttributes attr) { if (attr.on_host()) { return cpu_allocator(); } if (xla_allocator_ == nullptr) { if (UsePjRtForSingleDeviceCompilation(device_name_)) { VLOG(1) << "XlaDevice " << this << " uses AsyncValueAllocator"; pjrt_allocator_ = std::make_unique<AsyncValueAllocator>(); xla_allocator_ = pjrt_allocator_.get(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0)