chunk.js

let chunkSize = 15;
let inputWidth = 25;
let image.dims = {
  w: 1920,
  h: 1080
}

const myFunc = gpu.createKernel(
  function(imageTexture, offsetY, offsetX) {
    return imageTexture[this.thread.y + offsetY][this.thread.x + offsetX];
  },
  { output: [inputWidth, inputWidth] }
);

const chunks = [];
const imageTexture = new GPU.Texture(); // pseudo code, needs some inputs values
for (let yChunk = 0; yChunk < yChunks; yChunk++) {
  for (let xChunk = 0; xChunk < xChunks; xChunk++) {
    chunks.push({
      yChunk,
      xChunk,
      networkOutput: network.run(
        myFunc(
          imageTexture,
          yChunk * inputWidth,
          xChunk * inputWidth
        )
      ).toArray()
    });
  }
}