Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A 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
/aosp_15_r20/external/crosvm/devices/src/usb/xhci/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A 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()
H A 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
/aosp_15_r20/hardware/google/aemu/host-common/
Daddress_space_device.cpp120 AddressSpaceDeviceContext *device_context = contextDesc.device_context.get(); in ping() local
146 AddressSpaceDeviceContext *device_context = contextDesc.device_context.get(); in pingAtHva() local
206 const AddressSpaceDeviceContext *device_context = desc.device_context.get(); in save() local
224 const AddressSpaceDeviceContext *device_context = desc.device_context.get(); in save() local
244 const AddressSpaceDeviceContext *device_context = desc.device_context.get(); in save() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A 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()
H A Dpluggable_device.cc307 PluggableDeviceContext* device_context = device_context_; in ComputeAsync() local
428 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_util.cc23 RendezvousInterface* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
57 RendezvousInterface* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
H A Dthreadpool_device_test.cc58 DeviceContext* device_context = new DeviceContext; in TEST() local
H A Drenamed_device.h115 const DeviceContext* device_context, in CopyTensorInSameDevice()
H A 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
H A Dthreadpool_device.cc137 const DeviceContext* device_context, StatusCallback done) { in CopyTensorInSameDevice()
H A Dprocess_function_library_runtime_test.cc180 DeviceContext* device_context = in GPUToCPU() local
198 DeviceContext* device_context = in CPUToGPU() local
H A Dfunction.cc1039 DeviceContext* device_context; in RunRemote() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A 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
H A Dxla_launch_util.cc623 DeviceContext* device_context = nullptr; in BuildXlaCompilerArguments() local
/aosp_15_r20/external/autotest/client/cros/cellular/mbim_compliance/
H A Dentity.py15 def __init__(self, device_context): argument
H A Dmbim_data_transfer.py50 def __init__(self, device_context): argument
/aosp_15_r20/hardware/google/aemu/host-common/include/host-common/
DAddressSpaceService.h68 std::unique_ptr<AddressSpaceDeviceContext> device_context; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dunary_ops_composition_test.cc88 DeviceContext* device_context = in RunComposedOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drendezvous.h52 DeviceContext* device_context = nullptr; member
H A Ddevice_base.h275 const DeviceContext* device_context, in CopyTensorInSameDevice()
H A Dop_kernel_test.cc417 const DeviceContext* device_context, in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc446 auto device_context = in CallTfKernel() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc809 const DeviceContext* device_context, in CopyTensorInSameDevice()

12