Searched defs:GetOnDeviceSizeInBytes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | pjrt_c_api_client.cc | 686 StatusOr<size_t> PjRtCApiBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::PjRtCApiBuffer |
| D | tfrt_cpu_pjrt_client.cc | 662 StatusOr<size_t> TfrtCpuBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::TfrtCpuBuffer |
| D | pjrt_stream_executor_client.cc | 1378 StatusOr<size_t> PjRtStreamExecutorBuffer::GetOnDeviceSizeInBytes() const { in GetOnDeviceSizeInBytes() function in xla::PjRtStreamExecutorBuffer |