LOAD / STORE UNIT FOR A TENSOR ENGINE AND METHODS FOR LOADING OR STORING A TENSOR
A method for processing a tensor is described including obtaining a first register for a number of items in the tensor. One or more second registers for a number of items in a first and a second axis of the tensor are obtained. A stride in the first and the second axis is obtained A next item in the tensor is obtained using the stride in the first axis and a first offset register, when the first register indicates the tensor has additional items to process and the second registers indicate the next item resides in the first axis. A next item in the tensor is obtained using the stride in the first axis and the second axis, the first offset register, and a second offset register. The first register and a second register is modified. The first and the second offset registers are modified.
1 . A method for processing a tensor, comprising:
obtaining a first register for a number of items in the tensor;
obtaining one or more second registers for a number of items in a first and a second axis of the tensor;
obtaining a stride in the first and the second axis;
obtaining a next item in the tensor using the stride in the first axis and a first offset register, when the first register indicates the tensor has additional items to process and the second registers indicate the next item resides in the first axis;
obtaining a next item in the tensor using the stride in the first axis and the second axis, the first offset register, and a second offset register, when the first register indicates the tensor has additional items to process, and the second registers indicate the next item resides in the second axis of the tensor;
modifying the first register and one or more of the second registers; and
modifying at least one of the first and the second offset registers.