Home
last modified time | relevance | path

Searched refs:CopyToDevice (Results 1 – 25 of 26) sorted by relevance

12

/external/deqp-deps/amber/src/vulkan/
Dtransfer_buffer.h44 void CopyToDevice(CommandBuffer* command_buffer) override;
Dindex_buffer.cc49 transfer_buffer_->CopyToDevice(command); in SendIndexData()
Dframe_buffer.cc189 img->CopyToDevice(command); in TransferImagesToDevice()
192 depth_stencil_image_->CopyToDevice(command); in TransferImagesToDevice()
Dresource.h39 virtual void CopyToDevice(CommandBuffer* command_buffer) = 0;
Dtransfer_image.h54 void CopyToDevice(CommandBuffer* command_buffer) override;
Dtransfer_buffer.cc87 void TransferBuffer::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferBuffer
Dbuffer_backed_descriptor.cc55 resources[i]->CopyToDevice(command); in RecordCopyDataToResourceIfNeeded()
Dvertex_buffer.cc93 transfer_buffers_.back()->CopyToDevice(command); in SendVertexData()
Dtransfer_image.cc280 void TransferImage::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferImage
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcopy_to_device_node.h55 TF_RETURN_IF_ERROR(src_->CopyToDevice(ctx_, dstd_, &tensor)); in Run()
Dcore.cc75 *status = CopyToDevice(*ctx_, ctx_->HostCPU(), &tensor); in Resolve()
Dtensor_handle.h229 Status CopyToDevice(const EagerContext& ctx, tensorflow::Device* d,
Dtensor_handle.cc972 Status TensorHandle::CopyToDevice(const EagerContext& ctx, in CopyToDevice() function in tensorflow::TensorHandle
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.h68 StatusOr<std::unique_ptr<PyBuffer>> CopyToDevice(
Dpy_buffer.cc71 StatusOr<std::unique_ptr<PyBuffer>> PyBuffer::CopyToDevice( in CopyToDevice() function in xla::PyBuffer
78 TF_ASSIGN_OR_RETURN(out, buffer_->CopyToDevice(dst_device.get())); in CopyToDevice()
Dxla.cc306 .def("copy_to_device", &PyBuffer::CopyToDevice) in PYBIND11_MODULE()
Djax_jit.cc656 ValueOrThrow(buffer->buffer()->CopyToDevice(to_device)); in HandleDeviceArray()
/external/tensorflow/tensorflow/go/
Dtensor_handle.go161 func (th *TensorHandle) CopyToDevice(c *Context, deviceName string) (*TensorHandle, error) { func
/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.cc81 th->CopyToDevice(*context, /*d=*/nullptr, &inputs_storage.back())); in GetCompilerIr()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc322 TF_RETURN_IF_ERROR(h->CopyToDevice(*ctx, ctx->HostCPU(), &tensor)); in SerializePackedHandle()
432 s = src_->CopyToDevice(*ctx_, ctx_->HostCPU(), &tensor); in StartRemoteSendTensor()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h368 virtual StatusOr<std::unique_ptr<PjRtBuffer>> CopyToDevice(
Dpjrt_stream_executor_client.h518 StatusOr<std::unique_ptr<PjRtBuffer>> CopyToDevice(
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc150 return buffer->CopyToDevice(std::move(dst_device)); in PYBIND11_MODULE()
Dtpu_client.h237 StatusOr<std::unique_ptr<PyTpuBuffer>> CopyToDevice(
Dtpu_client.cc390 StatusOr<std::unique_ptr<PyTpuBuffer>> PyTpuBuffer::CopyToDevice( in CopyToDevice() function in xla::PyTpuBuffer

12