Searched defs:IsOnCpu (Results 1 – 2 of 2) sorted by relevance
461 bool IsOnCpu() const override { return true; } in IsOnCpu() function
577 bool PjRtStreamExecutorBuffer::IsOnCpu() const { in IsOnCpu() function in xla::PjRtStreamExecutorBuffer