Searched defs:IsOnCpu (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| H A D | tfrt_cpu_pjrt_client.h | 341 bool IsOnCpu() const override { return true; } in IsOnCpu() function |
| H A D | pjrt_c_api_client.cc | 747 bool PjRtCApiBuffer::IsOnCpu() const { in IsOnCpu() function in xla::PjRtCApiBuffer |
| H A D | pjrt_stream_executor_client.cc | 587 bool PjRtStreamExecutorBuffer::IsOnCpu() const { in IsOnCpu() function in xla::PjRtStreamExecutorBuffer |