import { a as l } from "../../index-DSGwv2Yx.js";
class u {
  variableNames = ["labels", "logits", "values"];
  outputShape;
  userCode;
  constructor(e) {
    this.outputShape = [e], this.userCode = `
          void main() {
              int coords = getOutputCoords();
              int index = int(getLabelsAtOutCoords());
              float val = getValuesAtOutCoords();
              float logit = getLogits(coords, index);
              setOutput(val - logit);
          }
    `;
  }
}
function i(t) {
  const { logits: e, labels: o, values: a } = t.inputs, s = t.backend, r = o.shape[0], n = new u(r);
  return s.runWebGLProgram(n, [o, e, a], "float32");
}
const c = {
  kernelName: "EfficientGatherSub",
  backendName: "webgl",
  kernelFunc: i
};
l(c);
