| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 39 const DeviceContext* device_context,
48 const DeviceContext* device_context,
81 const DeviceContext* device_context,
89 const DeviceContext* device_context,
94 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
104 const DeviceContext* device_context,
|
| D | gpu_util.cc | 61 Status PrepareCopy(Device* device, const DeviceContext* ctx, const Tensor& src, in PrepareCopy()
114 const DeviceContext* device_context, in SetProtoFromGPU()
188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.h | 32 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
43 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
44 DeviceContext* recv_dev_context, Device* src, Device* dst,
75 DeviceContext* send_dev_context, StatusCallback done);
|
| D | collective_rma_local.h | 42 DeviceContext* to_device_ctx,
51 DeviceContext* from_device_ctx,
66 static void MemCpyAsync(DeviceContext* src_dev_ctx,
67 DeviceContext* dst_dev_ctx, Device* src_dev,
|
| D | buf_rendezvous.h | 30 class DeviceContext; variable
65 DeviceContext* prod_ctx;
90 void ProvideBuf(const string& key, Device* dev, DeviceContext* dev_ctx,
|
| D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
99 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer()
119 DeviceContext* src_dev_ctx, DeviceContext* dst_dev_ctx, Device* src_dev, in MemCpyAsync()
|
| D | rendezvous_util.h | 34 RendezvousInterface* rendezvous, DeviceContext* device_context,
43 RendezvousInterface* rendezvous, DeviceContext* device_context,
|
| D | renamed_device.h | 100 DeviceContext* dc, in ReinitializeGpuDevice()
114 const DeviceContext* device_context, in CopyTensorInSameDevice()
137 Status TryGetDeviceContext(DeviceContext** out_context) override { in TryGetDeviceContext()
|
| D | copy_tensor.cc | 54 DeviceContext* recv_dev_context, StatusCallback done, in CopyHostToDevice()
122 DeviceContext* send_dev_context, in CopyDeviceToDevice()
123 DeviceContext* recv_dev_context, Device* src, in CopyDeviceToDevice()
200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()
201 DeviceContext* recv_dev_context, Device* src, in ViaDMA()
335 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost()
|
| D | threadpool_device_test.cc | 58 DeviceContext* device_context = new DeviceContext; in TEST()
|
| D | rendezvous_util.cc | 23 RendezvousInterface* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
57 RendezvousInterface* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
|
| D | threadpool_device.h | 42 const DeviceContext* device_context,
|
| D | collective_util.h | 42 gtl::InlinedVector<DeviceContext*, 4> sub_input_dc_;
|
| D | permuter_test.cc | 77 DeviceContext* to_device_ctx, in RecvFromPeer()
91 DeviceContext* from_device_ctx, in PostToPeer()
383 DeviceContext* dev_ctx = nullptr; in DoPermute()
389 dev_ctx = new DeviceContext; in DoPermute()
|
| D | process_function_library_runtime.h | 94 DeviceContext* device_context,
107 DeviceContext* device_context,
336 DeviceContext** device_context) const;
|
| D | single_threaded_cpu_device.cc | 73 const DeviceContext*, in CopyTensorInSameDevice() argument
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.cc | 48 class XRTMemoryManager::DeviceContext { class in tensorflow::XRTMemoryManager
219 DeviceContext* device_context = GetDeviceContext(tuple->device_ordinal(), in Register()
227 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Lookup()
241 DeviceContext* device_context = GetDeviceContext(device_ordinal, in Release()
251 DeviceContext* device_context = GetDeviceContext(device_ordinal, in CompactAllocations()
276 DeviceContext* device_context = in Allocate()
304 XRTMemoryManager::DeviceContext* XRTMemoryManager::GetDeviceContext( in GetDeviceContext()
313 DeviceContext* device_context = device_contexts_[device_ordinal].get(); in GetDeviceContext()
315 device_contexts_[device_ordinal] = absl::make_unique<DeviceContext>(); in GetDeviceContext()
323 DeviceContext* device_context = GetDeviceContext(mrctx->device_ordinal, in TryFreeMemoryStep()
|
| D | xrt_memory_manager.h | 45 class DeviceContext; variable
145 DeviceContext* GetDeviceContext(int device_ordinal, bool create_if_missing);
154 std::vector<std::unique_ptr<DeviceContext>> device_contexts_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 67 class DeviceContext : public core::RefCounted {
69 ~DeviceContext() override {} in ~DeviceContext()
156 DeviceContext* default_context = nullptr;
221 DeviceContext* /*dc*/, in ReinitializeGpuDevice() argument
263 const DeviceContext* device_context, in CopyTensorInSameDevice()
|
| D | device_base.cc | 37 Status DeviceContext::CopyDeviceTensorToCPUSync(const Tensor* device_tensor, in CopyDeviceTensorToCPUSync()
52 Status DeviceContext::CopyCPUTensorToDeviceSync(const Tensor* cpu_tensor, in CopyCPUTensorToDeviceSync()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 23 class VirtualDeviceContext : public DeviceContext {
92 Status VirtualDevice::TryGetDeviceContext(DeviceContext** out_context) { in TryGetDeviceContext()
|
| D | virtual_device.h | 34 Status TryGetDeviceContext(DeviceContext** out_context) override;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 320 DeviceContext* to_device_ctx = nullptr; in TEST_F()
350 DeviceContext* to_device_ctx = nullptr; in TEST_F()
385 DeviceContext* to_device_ctx = nullptr; in TEST_F()
413 DeviceContext* to_device_ctx = nullptr; in TEST_F()
|
| D | collective_rma_distributed.cc | 39 DeviceContext* to_device_ctx, in RecvBufCall()
80 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 42 DeviceContext* to_device_ctx,
|