Searched refs:dstd (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | copy_to_device_node.h | 29 CopyToDeviceNode(TensorHandle* src, TensorHandle* dst, Device* dstd, in CopyToDeviceNode() argument
31 : EagerNode(), src_(src), dst_(dst), dstd_(dstd), ctx_(ctx) { in CopyToDeviceNode()
|
| D | tensor_handle.cc | 594 tensorflow::Device* dstd, in CopyToDevice() argument
597 const bool dst_cpu = dstd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
600 (srcd == dstd) || (srcd->name() == dstd->name()) || (dst_cpu && src_cpu); in CopyToDevice()
612 tensorflow::DataTypeString(src->dtype()), " to device ", dstd->name(), in CopyToDevice()
619 tensorflow::Tensor dst(dstd->GetAllocator(attr), src->dtype(), src->shape()); in CopyToDevice()
630 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
643 srcd, dstd, tensorflow::AllocatorAttributes(), in CopyToDevice()
|
| D | execute.cc | 1009 EagerExecutor* executor, Device* dstd, in LocalEagerCopyToDevice() argument
1013 true, ctx->CanonicalDevice(dstd), dstd, h->resource_device(), h->dtype, in LocalEagerCopyToDevice()
1018 std::unique_ptr<EagerNode> node(new CopyToDeviceNode(h, *result, dstd, *ctx)); in LocalEagerCopyToDevice()
|
| D | tensor_handle.h | 170 Status CopyToDevice(const EagerContext& ctx, tensorflow::Device* dstd,
|