Searched refs:DeviceTensorAlloc (Results 1 – 2 of 2) sorted by relevance
37 using DeviceTensorAlloc = Allocator<DeviceTensor>; // An allocator for Device_Tensors variable89 …const DeviceTensorAlloc *device_tensor_allocator() const { return device_tensor_allocator_.get(); } in device_tensor_allocator()109 std::unique_ptr<DeviceTensorAlloc> device_tensor_allocator_; // An allocator for Device Tensors
41 const DeviceTensorAlloc *alloc = GlobalContext::Instance()->device_tensor_allocator(); in CreateEmpty()68 const DeviceTensorAlloc *alloc = GlobalContext::Instance()->device_tensor_allocator(); in CreateFromDeviceMemory()