Searched refs:GetDeviceContext (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.cc | 219 DeviceContext* device_context = GetDeviceContext(tuple->device_ordinal(), in Register()
227 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Lookup()
241 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Release()
251 DeviceContext* device_context = GetDeviceContext(device_ordinal, in CompactAllocations()
277 GetDeviceContext(device_ordinal, in Allocate()
304 XRTMemoryManager::DeviceContext* XRTMemoryManager::GetDeviceContext( in GetDeviceContext() function in tensorflow::XRTMemoryManager
323 DeviceContext* device_context = GetDeviceContext(mrctx->device_ordinal, in TryFreeMemoryStep()
|
| D | xrt_memory_manager.h | 145 DeviceContext* GetDeviceContext(int device_ordinal, bool create_if_missing);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 335 Status GetDeviceContext(const string& device_name,
|
| D | process_function_library_runtime.cc | 185 Status ProcessFunctionLibraryRuntime::GetDeviceContext( in GetDeviceContext() function in tensorflow::ProcessFunctionLibraryRuntime
1507 Status s = GetDeviceContext(source_device, &device_context); in RunInternal()
|
| D | function.cc | 1005 Status s = parent_->GetDeviceContext(target_device, &device_context); in RunRemote()
|