Home
last modified time | relevance | path

Searched refs:CreateViewOfDeviceBuffer (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h244 virtual StatusOr<std::unique_ptr<PjRtBuffer>> CreateViewOfDeviceBuffer(
Dpjrt_stream_executor_client.h200 StatusOr<std::unique_ptr<PjRtBuffer>> CreateViewOfDeviceBuffer(
Dpjrt_stream_executor_client.cc958 PjRtStreamExecutorClient::CreateViewOfDeviceBuffer( in CreateViewOfDeviceBuffer() function in xla::PjRtStreamExecutorClient
/external/tensorflow/tensorflow/compiler/xla/python/
Ddlpack.cc359 client->pjrt_client()->CreateViewOfDeviceBuffer( in DLPackManagedTensorToBuffer()