Home
last modified time | relevance | path

Searched defs:device_context (Results 1 – 25 of 25) 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()
267 const DeviceContext* device_context, in CopyGPUTensorToCPU()
312 const DeviceContext* device_context, in CopyCPUTensorToGPU()
440 const DeviceContext* device_context, in Checksum()
469 const DeviceContext* device_context, in CopyGPUTensorToSameGPU()
Dgpu_device_test.cc115 DeviceContext* device_context) { in CopyCPUToGPU()
121 DeviceContext* device_context) { in CopyGPUToCPU()
527 DeviceContext* device_context = device_info->default_context; in TEST_F() local
Dgpu_device.cc893 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/autotest/client/cros/cellular/mbim_compliance/
Dentity.py15 def __init__(self, device_context): argument
Dmbim_data_transfer.py50 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
Drenamed_device.h115 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dthreadpool_device.cc137 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
Dprocess_function_library_runtime.cc123 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors()
142 DeviceContext* device_context, in ReceiveTensorsAsync()
1796 DeviceContext* device_context; in RunInternal() local
Dprocess_function_library_runtime_test.cc180 DeviceContext* device_context = in GPUToCPU() local
198 DeviceContext* device_context = in CPUToGPU() local
Dfunction.cc1039 DeviceContext* device_context; in RunRemote() local
/external/crosvm/devices/src/usb/xhci/
Ddevice_slot.rs427 let mut device_context = slot.get_device_context()?; in disable() localVariable
458 let device_context = self.get_device_context()?; in set_address() localVariable
481 let mut device_context = self.get_device_context()?; in set_address() localVariable
601 let mut device_context = self.get_device_context()?; in evaluate_context() localVariable
694 let mut device_context = self.get_device_context()?; in stop_endpoint() localVariable
758 let mut device_context = self.get_device_context()?; in reset_endpoint() localVariable
902 let mut device_context = self.get_device_context()?; in add_one_endpoint() localVariable
968 let mut device_context = self.get_device_context()?; in drop_one_endpoint() localVariable
1053 let mut device_context = self.get_device_context()?; in halt_endpoint() localVariable
/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/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc370 auto device_context = new XlaDeviceContext( in GetDeviceContextLocked() local
504 Status XlaDevice::MakeTensorFromProto(XlaDeviceContext* device_context, in MakeTensorFromProto()
536 XlaDeviceContext* device_context; in MakeTensorFromProto() local
Dxla_launch_util.cc623 DeviceContext* device_context = nullptr; in BuildXlaCompilerArguments() local
/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.h275 const DeviceContext* device_context, in CopyTensorInSameDevice()
Dop_kernel_test.cc417 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlight_outside_compilation.cc446 auto device_context = in CallTfKernel() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute_compat.cc809 const DeviceContext* device_context, in CopyTensorInSameDevice()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1651 DeviceContext* device_context = dev_info->default_context; in CreateContext() local