Home
last modified time | relevance | path

Searched refs:BufferFromHostBuffer (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc74 client->BufferFromHostBuffer( in TEST()
80 client->BufferFromHostBuffer( in TEST()
86 client->BufferFromHostBuffer( in TEST()
Dpjrt_client.h225 virtual StatusOr<std::unique_ptr<PjRtBuffer>> BufferFromHostBuffer(
Dpjrt_stream_executor_client.h187 StatusOr<std::unique_ptr<PjRtBuffer>> BufferFromHostBuffer(
Dpjrt_stream_executor_client.cc647 PjRtStreamExecutorClient::BufferFromHostBuffer( in BufferFromHostBuffer() function in xla::PjRtStreamExecutorClient
1368 return dst_device->client()->BufferFromHostBuffer( in CopyToDevice()
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.cc140 pjrt_client_->BufferFromHostBuffer( in PjRtBufferFromPyval()
Djax_jit.cc367 return ValueOrThrow(client->BufferFromHostBuffer( in ConvertToScalarBuffer()
623 ValueOrThrow(pyclient.pjrt_client()->BufferFromHostBuffer( in HandleComplex()
632 ValueOrThrow(pyclient.pjrt_client()->BufferFromHostBuffer( in HandleComplex()