@tensorflow/tfjs-core/dist-es6/kernels/webgl/onehot_gpu.js

Hardware-accelerated JavaScript library for machine intelligence

var OneHotProgram = (function () {
    function OneHotProgram(numIndices, depth, onValue, offValue) {
        this.variableNames = ['indices'];
        this.outputShape = [numIndices, depth];
        this.userCode = "\n      void main() {\n        ivec2 coords = getOutputCoords();\n        int index = round(getIndices(coords.x));\n        setOutput(mix(float(" + offValue + "), float(" + onValue + "),\n                      float(index == coords.y)));\n      }\n    ";
    }
    return OneHotProgram;
}());
export { OneHotProgram };
//# sourceMappingURL=onehot_gpu.js.map