Method and apparatus for generating architecture specific convolution gradient kernels

A method for accelerating a convolution operation, the method comprising:

• receiving, by a task scheduler from an input/output (I/O) interface, a first data set and a second data set;
• transforming, by the task scheduler, the first data set into a first converted data set, the first converted data set having the first format;
• transforming, by the task scheduler, the second data set into a second converted data set, the second converted data set having the second format;
• loading, by the task scheduler into a convolution functional unit, the first converted data set and the second converted data set, the convolution functional unit configured to receive a first data in a first format, to receive a second data in a second format, and to output a third data in a third format;
• and
• receiving, by the task scheduler from the convolution functional unit, a result of the convolution operation in the third format.

A method for transforming tensors, the method comprising:

• receiving an input tensor having an input format, the input format having a set of input iterators, and storing the input tensor into an input memory;
• allocating an output memory to store an output tensor having an output format, the output format having a set of output iterators;
• and
• iterating over the output memory to copy an input component of the input tensor to a portion of the output memory corresponding to an output component of the output tensor, an index of the input component defined as a function of the output set of iterators.