Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc686 StatusOr<size_t> PjRtCApiBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::PjRtCApiBuffer
H A Dtfrt_cpu_pjrt_client.cc662 StatusOr<size_t> TfrtCpuBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::TfrtCpuBuffer
H A Dpjrt_stream_executor_client.cc1378 StatusOr<size_t> PjRtStreamExecutorBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::PjRtStreamExecutorBuffer