Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h182 StatusOr<std::unique_ptr<PjRtBuffer>> CreateUninitializedBuffer( in CreateUninitializedBuffer() function
H A Dpjrt_stream_executor_client.cc907 PjRtStreamExecutorClient::CreateUninitializedBuffer(const Shape& shape, in CreateUninitializedBuffer() function in xla::PjRtStreamExecutorClient
913 PjRtStreamExecutorClient::CreateUninitializedBuffer( in CreateUninitializedBuffer() function in xla::PjRtStreamExecutorClient
H A Dtfrt_cpu_pjrt_client.cc480 StatusOr<std::unique_ptr<PjRtBuffer>> TfrtCpuClient::CreateUninitializedBuffer( in CreateUninitializedBuffer() function in xla::TfrtCpuClient