Devices and methods for performing computations of a neural network include determining, for a particular layer of the neural network, a number of input channels and a number of output maps generated for particular pixels of a plurality of pixels. A portion of a processing chip is configured into a plurality of processing units, with the processing chip including a plurality of tensor arrays and a plurality memory cells. A particular processing unit of the plurality of processing units performs computations associated with a particular pixel. Individual processing units of the plurality of processing units each include a number of tensor arrays determined based on the number of input channels and a number of memory cells or pixel arrays corresponding to a number of output maps. The plurality of processing units is assigned to perform computations of the particular layer.