| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 116 const DeviceContext* device_context, in SetProtoFromGPU()
256 const DeviceContext* device_context, in CopyGPUTensorToCPU()
301 const DeviceContext* device_context, in CopyCPUTensorToGPU()
394 const DeviceContext* device_context, in Checksum()
423 const DeviceContext* device_context, in CopyGPUTensorToSameGPU()
|
| D | gpu_device.cc | 602 DeviceContext* device_context, const TensorReferenceVector& tensor_refs) { in ConsumeListOfAccessedTensors()
|
| /external/autotest/client/cros/cellular/mbim_compliance/ |
| D | entity.py | 15 def __init__(self, device_context): argument
|
| D | mbim_data_transfer.py | 42 def __init__(self, device_context): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_util.cc | 23 Rendezvous* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
57 Rendezvous* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
|
| D | process_function_library_runtime.cc | 113 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors()
132 DeviceContext* device_context, in ReceiveTensorsAsync()
950 DeviceContext* device_context; in Run() local
|
| D | process_function_library_runtime_test.cc | 109 DeviceContext* device_context = in GPUToCPU() local
133 DeviceContext* device_context = in CPUToGPU() local
|
| D | executor.cc | 935 DeviceContext* device_context = nullptr; member
1705 DeviceContext* device_context = nullptr; in Process() local
2005 DeviceContext* device_context = nullptr; in ProcessOutputs() local
|
| D | function.cc | 956 DeviceContext* device_context; in RunRemote() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 49 DeviceContext* device_context = nullptr; member
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 367 Device* device, DeviceContext* device_context, bool on_host, in TransportOptionsFromTensor()
425 Device* device, DeviceContext* device_context, bool on_host, in TensorFromTransportOptions()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | mpi_ops.cc | 1012 auto device_context = context->op_device_context(); in ComputeAsync() local
1106 auto device_context = context->op_device_context(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops.cc | 1066 auto device_context = context->op_device_context(); in ComputeAsync() local
1182 auto device_context = context->op_device_context(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 224 DeviceContext* device_context = nullptr; in Run() local
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 913 static uint64_t Checksum(Device* device, const DeviceContext* device_context, in Checksum()
964 static void StreamGPUOp(Device* gpu_device, const DeviceContext* device_context, in StreamGPUOp()
|