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.cc222 DeviceContext* device_context = GetDeviceContext(tuple->device_ordinal(), in Register() local
230 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Lookup() local
244 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Release() local
255 DeviceContext* device_context = GetDeviceContext(device_ordinal, in CompactAllocations() local
264 for (auto& device_context : device_contexts_) { in ReleaseAllAllocations() local
280 DeviceContext* device_context = in Allocate() local
317 DeviceContext* device_context = device_contexts_[device_ordinal].get(); in GetDeviceContext() local
327 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.cc100 DeviceContext* device_context) { in CopyCPUToGPU()
106 DeviceContext* device_context) { in CopyGPUToCPU()
406 DeviceContext* device_context = device_info->default_context; in TEST_F() local
Dgpu_device.cc862 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
Drenamed_device.h114 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dthreadpool_device.cc126 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
Dprocess_function_library_runtime.cc120 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors()
139 DeviceContext* device_context, in ReceiveTensorsAsync()
1521 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.cc1098 DeviceContext* device_context; in RunRemote() 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/pluggable_device/
Dpluggable_device_util.cc175 Device* device, const DeviceContext* device_context, in CopyPluggableDeviceTensorToCPU()
222 const Tensor* cpu_tensor, const DeviceContext* device_context, in CopyCPUTensorToPluggableDevice()
293 Device* device, const DeviceContext* device_context, in CopyPluggableDeviceTensorToSameDevice()
Dpluggable_device.cc307 PluggableDeviceContext* device_context = device_context_; in ComputeAsync() local
428 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
/external/crosvm/devices/src/usb/xhci/
Ddevice_slot.rs363 let mut device_context = slot.get_device_context()?; in disable() localVariable
391 let device_context = self.get_device_context()?; in set_address() localVariable
414 let mut device_context = self.get_device_context()?; in set_address() localVariable
538 let mut device_context = self.get_device_context()?; in evaluate_context() localVariable
681 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.cc504 Status XlaDevice::MakeTensorFromProto(XlaDeviceContext* device_context, in MakeTensorFromProto()
Dxla_launch_util.cc619 DeviceContext* device_context = nullptr; in BuildXlaCompilerArguments() local