Home
last modified time | relevance | path

Searched refs:IsOnCpu (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.cc92 if (!buffer_->IsOnCpu() && !host_value_) { in CopyToHostAsync()
114 if (buffer_->IsOnCpu()) { in AsNumPyArray()
230 if (!buffer.IsOnCpu()) { in PjRtBufferGetBuffer()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h388 virtual bool IsOnCpu() const = 0;
Dpjrt_stream_executor_client.h525 bool IsOnCpu() const override;
Dpjrt_stream_executor_client.cc577 bool PjRtStreamExecutorBuffer::IsOnCpu() const { in IsOnCpu() function in xla::PjRtStreamExecutorBuffer