Home
last modified time | relevance | path

Searched defs:device_context (Results 1 – 22 of 22) 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/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/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.cc96 DeviceContext* device_context) { in CopyCPUToGPU()
107 DeviceContext* device_context) { in CopyGPUToCPU()
317 DeviceContext* device_context = device_info->default_context; in TEST_F() local
Dgpu_device.cc545 DeviceContext* device_context, const TensorReferenceVector& tensor_refs) { in ConsumeListOfAccessedTensors()
697 const DeviceContext* device_context, in CopyTensorInSameDevice()
/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.cc108 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
Drenamed_device.h124 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dprocess_function_library_runtime.cc124 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors()
143 DeviceContext* device_context, in ReceiveTensorsAsync()
1303 DeviceContext* device_context; in RunInternal() local
Dprocess_function_library_runtime_test.cc139 DeviceContext* device_context = in GPUToCPU() local
163 DeviceContext* device_context = in CPUToGPU() local
Dfunction.cc1022 DeviceContext* device_context; in RunRemote() local
Dexecutor.cc1796 DeviceContext* device_context = nullptr; in Process() local
/external/crosvm/devices/src/usb/xhci/
Ddevice_slot.rs366 let mut device_context = slot.get_device_context()?; in disable() localVariable
394 let device_context = self.get_device_context()?; in set_address() localVariable
417 let mut device_context = self.get_device_context()?; in set_address() localVariable
541 let mut device_context = self.get_device_context()?; in evaluate_context() localVariable
684 let mut device_context = self.get_device_context()?; in add_one_endpoint() localVariable
/external/tensorflow/tensorflow/c/experimental/
Drendezvous.cc59 TF_DeviceContext* device_context = new TF_DeviceContext(); in RecvFromRemoteAsync() local
Drendezvous_internal.h56 const TF_DeviceContext* device_context; member
/external/tensorflow/tensorflow/core/framework/
Drendezvous.h50 DeviceContext* device_context = nullptr; member
Ddevice_base.h273 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dop_kernel_test.cc470 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc82 DeviceContext* device_context = in RunComposedOp() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc474 Status XlaDevice::MakeTensorFromProto(XlaDeviceContext* device_context, in MakeTensorFromProto()