Function getInputSnippet

tfjs-backend-webgpu/src/webgpu_program.ts:627–639

(
    inputInfo: InputInfo, outShape: number[], component: number,
    isFlatDispatchLayout: boolean)

}

function getInputSnippet(
    inputInfo: InputInfo, outShape: number[], component: number,
    isFlatDispatchLayout: boolean): string {
  let res = getInputAtCoordsSnippet(inputInfo, component);

  const inShape = inputInfo.shape;
  if (inShape.length <= outShape.length) {
    res += getInputByOutputSnippet(
        inputInfo, outShape, component, isFlatDispatchLayout);
  }

  return res;
}

/**
 * Generates getOutputCoords() function that computes output coordinates