Max pooling in a matrix processing architecture

A matrix processing circuit, comprising:

• a plurality of memory resource blocks;
• control circuitry to cause the matrix processing circuit to perform a backward pooling operation, wherein the backward pooling operation is to generate, based on a pooled matrix operand, a partially reconstructed matrix corresponding to an original matrix operand, wherein the pooled matrix operand corresponds to an output of a forward pooling operation performed on the original matrix operand;
• read circuitry to receive the pooled matrix operand from one or more of the plurality of memory resource blocks;
• slice circuitry to:
• extract a plurality of value-index pairs from the pooled matrix operand;
• and
• for each value-index pair of the plurality of value-index pairs:
• generate, based on the value-index pair, a corresponding portion of the partially reconstructed matrix;
• identify a completed portion of the partially reconstructed matrix to write to memory, wherein the completed portion will not be modified during generation of a remaining portion of the partially reconstructed matrix;
• and
• write the completed portion of the partially reconstructed matrix to one or more of the plurality of memory resource blocks;
• and
• output circuitry to output the partially reconstructed matrix as a result of the backward pooling operation.

A system, comprising:

• a processor to execute a plurality of instructions associated with an application, wherein the plurality of instructions comprises one or more instructions associated with a backward pooling operation, wherein the backward pooling operation is to generate, based on a pooled matrix operand, a partially reconstructed matrix corresponding to an original matrix operand, wherein the pooled matrix operand corresponds to an output of a forward pooling operation performed on the original matrix operand;
• and
• matrix processing circuitry to perform the backward pooling operation, wherein the matrix processing circuitry comprises:
• a plurality of memory resource blocks;
• and
• circuitry to:
• receive the pooled matrix operand from one or more of the plurality of memory resource blocks;
• extract a plurality of value-index pairs from the pooled matrix operand;
• for each value-index pair of the plurality of value-index pairs:
• generate, based on the value-index pair, a corresponding portion of the partially reconstructed matrix;
• identify a completed portion of the partially reconstructed matrix to write to memory, wherein the completed portion will not be modified during generation of a remaining portion of the partially reconstructed matrix;
• and
• write the completed portion of the partially reconstructed matrix to one or more of the plurality of memory resource blocks;
• and
• output the partially reconstructed matrix as a result of the backward pooling operation.

At least one non-transitory machine accessible storage medium having instructions stored thereon, the instructions, when executed on a machine, cause the machine to:

• receive a command to perform a backward pooling operation, wherein the backward pooling operation is to generate, based on a pooled matrix operand, a partially reconstructed matrix corresponding to an original matrix operand, wherein the pooled matrix operand corresponds to an output of a forward pooling operation performed on the original matrix operand;
• receive the pooled matrix operand from one or more of a plurality of memory resource blocks;
• extract a plurality of value-index pairs from the pooled matrix operand;
• for each value-index pair of the plurality of value-index pairs:
• generate, based on the value-index pair, a corresponding portion of the partially reconstructed matrix;
• identify a completed portion of the partially reconstructed matrix to write to memory, wherein the completed portion will not be modified during generation of a remaining portion of the partially reconstructed matrix;
• and
• write the completed portion of the partially reconstructed matrix to one or more of the plurality of memory resource blocks;
• and
• output the partially reconstructed matrix as a result of the backward pooling operation.

A method, comprising:

• receiving a command to perform a backward pooling operation, wherein the backward pooling operation is to generate, based on a pooled matrix operand, a partially reconstructed matrix corresponding to an original matrix operand, wherein the pooled matrix operand corresponds to an output of a forward pooling operation performed on the original matrix operand;
• receiving the pooled matrix operand from one or more of a plurality of memory resource blocks;
• extracting a plurality of value-index pairs from the pooled matrix operand;
• for each value-index pair of the plurality of value-index pairs:
• generating, based on the value-index pair, a corresponding portion of the partially reconstructed matrix;
• identifying a completed portion of the partially reconstructed matrix to write to memory, wherein the completed portion will not be modified during generation of a remaining portion of the partially reconstructed matrix;
• and
• writing the completed portion of the partially reconstructed matrix to one or more of the plurality of memory resource blocks;
• and
• outputting the partially reconstructed matrix as a result of the backward pooling operation.