Searched defs:logical_on_device_shape (Results 1 – 3 of 3) sorted by relevance
311 StatusOr<Shape> logical_on_device_shape() override { in logical_on_device_shape() function
860 StatusOr<Shape> TfrtCpuBuffer::logical_on_device_shape() { in logical_on_device_shape() function in xla::TfrtCpuBuffer
591 StatusOr<Shape> PjRtStreamExecutorBuffer::logical_on_device_shape() { in logical_on_device_shape() function in xla::PjRtStreamExecutorBuffer