UNPKG

keras-js

Version:

Run Keras models in the browser, with GPU support using WebGL

65 lines (42 loc) 1.74 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.default = createGLSLProgram; var _add = _interopRequireDefault(require("./merge/add")); var _average = _interopRequireDefault(require("./merge/average")); var _concatenate = _interopRequireDefault(require("./merge/concatenate")); var _maximum = _interopRequireDefault(require("./merge/maximum")); var _minimum = _interopRequireDefault(require("./merge/minimum")); var _multiply = _interopRequireDefault(require("./merge/multiply")); var _subtract = _interopRequireDefault(require("./merge/subtract")); var _conv2d = _interopRequireDefault(require("./convolutional/conv2d")); var _conv2dTranspose = _interopRequireDefault(require("./convolutional/conv2dTranspose")); var _cam = _interopRequireDefault(require("./visualizations/cam")); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function createGLSLProgram(program, ...args) { switch (program) { case 'add': return (0, _add.default)(...args); case 'average': return (0, _average.default)(...args); case 'concatenate': return (0, _concatenate.default)(...args); case 'maximum': return (0, _maximum.default)(...args); case 'minimum': return (0, _minimum.default)(...args); case 'multiply': return (0, _multiply.default)(...args); case 'subtract': return (0, _subtract.default)(...args); case 'conv2d': return (0, _conv2d.default)(...args); case 'conv2dTranspose': return (0, _conv2dTranspose.default)(...args); case 'cam': return (0, _cam.default)(...args); default: throw new Error('GLSL program not found'); } }