Home
last modified time | relevance | path

Searched refs:CreateUninitializedBuffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h181 StatusOr<std::unique_ptr<PjRtBuffer>> CreateUninitializedBuffer(
183 StatusOr<std::unique_ptr<PjRtBuffer>> CreateUninitializedBuffer(
Dpjrt_client.h193 virtual StatusOr<std::unique_ptr<PjRtBuffer>> CreateUninitializedBuffer(
Dpjrt_stream_executor_client.cc828 PjRtStreamExecutorClient::CreateUninitializedBuffer(const Shape& shape, in CreateUninitializedBuffer() function in xla::PjRtStreamExecutorClient
830 return CreateUninitializedBuffer(shape, device, nullptr); in CreateUninitializedBuffer()
834 PjRtStreamExecutorClient::CreateUninitializedBuffer( in CreateUninitializedBuffer() function in xla::PjRtStreamExecutorClient