Home
last modified time | relevance | path

Searched refs:dstd (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc251 Status TensorHandle::CopyToDevice(EagerContext* ctx, tensorflow::Device* dstd, in CopyToDevice() argument
260 bool is_same_device = (srcd == dstd) || (srcd->name() == dstd->name()); in CopyToDevice()
261 const bool dst_cpu = dstd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
264 *output = new tensorflow::TensorHandle(*src, dstd, dstd, ctx); in CopyToDevice()
271 tensorflow::DataTypeString(src->dtype()), " to device ", dstd->name(), in CopyToDevice()
278 tensorflow::Tensor dst(dstd->GetAllocator(attr), src->dtype(), src->shape()); in CopyToDevice()
280 dstd = dst_cpu ? nullptr : dstd; in CopyToDevice()
281 *output = new tensorflow::TensorHandle(dst, dstd, dstd, ctx); in CopyToDevice()
290 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
303 srcd, dstd, tensorflow::AllocatorAttributes(), in CopyToDevice()
[all …]
Dcopy_to_device_node.h28 CopyToDeviceNode(TensorHandle* src, Device* dstd, EagerContext* ctx) in CopyToDeviceNode() argument
31 dstd_(dstd), in CopyToDeviceNode()
Dtensor_handle.h105 Status CopyToDevice(EagerContext* ctx, tensorflow::Device* dstd,
Dexecute.cc1035 Status LocalEagerCopyToDevice(TensorHandle* h, EagerContext* ctx, Device* dstd, in LocalEagerCopyToDevice() argument
1041 CopyToDeviceNode* node = new CopyToDeviceNode(h, dstd, ctx); in LocalEagerCopyToDevice()
1049 TF_RETURN_IF_ERROR(h->CopyToDevice(ctx, dstd, result)); in LocalEagerCopyToDevice()