Load/Store Unit for a Tensor Engine and Methods for Loading or Storing a Tensor

A load/store unit (LDSU) for a tensor engine comprising:

• a first register for storing values associated with a number of items in a tensor;
• a plurality of second registers for storing a number of items in a first and a second axis of the tensor;
• a plurality of offset registers associated with the first and the second axis;
• a first and a second stride register associated with the first and the second axis;
• a tensor walking module configured to obtain a next item in the tensor using a first stride register 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 of the tensor,
• the tensor walking module further configured to obtain a next item in the tensor using the first and the second stride registers, 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;
• an iteration tracking module configured to modify the first and the second registers;
• and
• a striding module configured to modify at least one of the first offset register or the second offset register.