Searched defs:CopyToDevice (Results 1 – 12 of 12) sorted by relevance
93 void TransferBuffer::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferBuffer
280 void TransferImage::CopyToDevice(CommandBuffer* command_buffer) { in CopyToDevice() function in amber::vulkan::TransferImage
153 func (th *TensorHandle) CopyToDevice(c *Context, deviceName string) (*TensorHandle, error) { func
37 torch::Tensor CopyToDevice( in CopyToDevice() function
336 TEST(PjRtClientTest, CopyToDevice) { in TEST() argument
727 StatusOr<std::unique_ptr<PjRtBuffer>> PjRtCApiBuffer::CopyToDevice( in CopyToDevice() function in xla::PjRtCApiBuffer
1015 StatusOr<std::unique_ptr<PjRtBuffer>> TfrtCpuBuffer::CopyToDevice( in CopyToDevice() function in xla::TfrtCpuBuffer
1477 StatusOr<std::unique_ptr<PjRtBuffer>> PjRtStreamExecutorBuffer::CopyToDevice( in CopyToDevice() function in xla::PjRtStreamExecutorBuffer
188 StatusOr<py::object> PyBuffer::CopyToDevice( in CopyToDevice() function in xla::PyBuffer
401 StatusOr<std::unique_ptr<PyTpuBuffer>> PyTpuBuffer::CopyToDevice( in CopyToDevice() function in xla::PyTpuBuffer
908 def CopyToDevice(self, src, dest, mode, **kwargs): member in RemoteDevice
972 Status TensorHandle::CopyToDevice(const EagerContext& ctx, in CopyToDevice() function in tensorflow::TensorHandle