Searched defs:CopyToDevice (Results 1 – 7 of 7) sorted by relevance
87 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
161 func (th *TensorHandle) CopyToDevice(c *Context, deviceName string) (*TensorHandle, error) { func
71 StatusOr<std::unique_ptr<PyBuffer>> PyBuffer::CopyToDevice( in CopyToDevice() function in xla::PyBuffer
390 StatusOr<std::unique_ptr<PyTpuBuffer>> PyTpuBuffer::CopyToDevice( in CopyToDevice() function in xla::PyTpuBuffer
972 Status TensorHandle::CopyToDevice(const EagerContext& ctx, in CopyToDevice() function in tensorflow::TensorHandle
1354 StatusOr<std::unique_ptr<PjRtBuffer>> PjRtStreamExecutorBuffer::CopyToDevice( in CopyToDevice() function in xla::PjRtStreamExecutorBuffer