Home
last modified time | relevance | path

Searched defs:device_context (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_memory_manager.cc219 DeviceContext* device_context = GetDeviceContext(tuple->device_ordinal(), in Register() local
227 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Lookup() local
241 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Release() local
251 DeviceContext* device_context = GetDeviceContext(device_ordinal, in CompactAllocations() local
260 for (auto& device_context : device_contexts_) { in ReleaseAllAllocations() local
276 DeviceContext* device_context = in Allocate() local
313 DeviceContext* device_context = device_contexts_[device_ordinal].get(); in GetDeviceContext() local
323 DeviceContext* device_context = GetDeviceContext(mrctx->device_ordinal, in TryFreeMemoryStep() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc114 const DeviceContext* device_context, in SetProtoFromGPU()
255 const DeviceContext* device_context, in CopyGPUTensorToCPU()
300 const DeviceContext* device_context, in CopyCPUTensorToGPU()
394 const DeviceContext* device_context, in Checksum()
423 const DeviceContext* device_context, in CopyGPUTensorToSameGPU()
Dgpu_device_test.cc105 DeviceContext* device_context) { in CopyCPUToGPU()
111 DeviceContext* device_context) { in CopyGPUToCPU()
415 DeviceContext* device_context = device_info->default_context; in TEST_F() local
Dgpu_device.cc754 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/autotest/client/cros/cellular/mbim_compliance/
Dentity.py15 def __init__(self, device_context): argument
Dmbim_data_transfer.py42 def __init__(self, device_context): argument
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_util.cc23 RendezvousInterface* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
57 RendezvousInterface* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
Dthreadpool_device_test.cc58 DeviceContext* device_context = new DeviceContext; in TEST() local
Dthreadpool_device.cc104 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
Drenamed_device.h114 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dprocess_function_library_runtime.cc120 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors()
139 DeviceContext* device_context, in ReceiveTensorsAsync()
1506 DeviceContext* device_context; in RunInternal() local
Dprocess_function_library_runtime_test.cc178 DeviceContext* device_context = in GPUToCPU() local
196 DeviceContext* device_context = in CPUToGPU() local
Dfunction.cc1004 DeviceContext* device_context; in RunRemote() local
/external/crosvm/devices/src/usb/xhci/
Ddevice_slot.rs370 let mut device_context = slot.get_device_context()?; in disable() localVariable
398 let device_context = self.get_device_context()?; in set_address() localVariable
421 let mut device_context = self.get_device_context()?; in set_address() localVariable
545 let mut device_context = self.get_device_context()?; in evaluate_context() localVariable
688 let mut device_context = self.get_device_context()?; in add_one_endpoint() localVariable
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc88 DeviceContext* device_context = in RunComposedOp() local
/external/tensorflow/tensorflow/core/framework/
Drendezvous.h52 DeviceContext* device_context = nullptr; member
Ddevice_base.h263 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dop_kernel_test.cc414 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc474 Status XlaDevice::MakeTensorFromProto(XlaDeviceContext* device_context, in MakeTensorFromProto()
Dxla_launch_util.cc588 DeviceContext* device_context = nullptr; in BuildXlaCompilerArguments() local