Home
last modified time | relevance | path

Searched defs:CopyToDevice (Results 1 – 12 of 12) sorted by relevance

/external/deqp-deps/amber/src/vulkan/
Dtransfer_buffer.cc103 void TransferBuffer::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferBuffer
Dtransfer_image.cc280 void TransferImage::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferImage
/external/tensorflow/tensorflow/go/
Dtensor_handle.go153 func (th *TensorHandle) CopyToDevice(c *Context, deviceName string) (*TensorHandle, error) { func
/external/pytorch/test/cpp/lazy/
Dtest_lazy_ops_util.cpp37 torch::Tensor CopyToDevice( in CopyToDevice() function
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client_test.cc336 TEST(PjRtClientTest, CopyToDevice) { in TEST() argument
Dpjrt_c_api_client.cc727 StatusOr<std::unique_ptr<PjRtBuffer>> PjRtCApiBuffer::CopyToDevice( in CopyToDevice() function in xla::PjRtCApiBuffer
Dtfrt_cpu_pjrt_client.cc1015 StatusOr<std::unique_ptr<PjRtBuffer>> TfrtCpuBuffer::CopyToDevice( in CopyToDevice() function in xla::TfrtCpuBuffer
Dpjrt_stream_executor_client.cc1477 StatusOr<std::unique_ptr<PjRtBuffer>> PjRtStreamExecutorBuffer::CopyToDevice( in CopyToDevice() function in xla::PjRtStreamExecutorBuffer
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.cc188 StatusOr<py::object> PyBuffer::CopyToDevice( in CopyToDevice() function in xla::PyBuffer
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc401 StatusOr<std::unique_ptr<PyTpuBuffer>> PyTpuBuffer::CopyToDevice( in CopyToDevice() function in xla::PyTpuBuffer
/external/autotest/utils/frozen_chromite/lib/
Dremote_access.py908 def CopyToDevice(self, src, dest, mode, **kwargs): member in RemoteDevice
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc972 Status TensorHandle::CopyToDevice(const EagerContext& ctx, in CopyToDevice() function in tensorflow::TensorHandle