Searched refs:logical_on_device_shape (Results 1 – 7 of 7) sorted by relevance
311 StatusOr<Shape> logical_on_device_shape() override { in logical_on_device_shape() function313 return wrapped_->logical_on_device_shape(); in logical_on_device_shape()
297 StatusOr<Shape> logical_on_device_shape() override;
809 virtual StatusOr<Shape> logical_on_device_shape() = 0;
567 StatusOr<Shape> logical_on_device_shape() override;
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
161 TF_ASSIGN_OR_RETURN(dynamic_shape, buffer_->logical_on_device_shape()); in xla_dynamic_shape()