@hoff97/tensor-js/dist/lib/ops/gpu/binary/multiply.js

PyTorch like deep learning inferrence library

import { BinaryOperation } from './binaryOperation';
export class MultiplyOperation extends BinaryOperation {
    constructor(tensorConstructor, dtype, allocator) {
        super(tensorConstructor, dtype, allocator);
    }
    getOp(a, b) {
        return `alpha*${a} * ${b}`;
    }
    calc(input) {
        return this.compute(input.outputShape, { A: input.A, B: input.B }, { alpha: input.alpha });
    }
    getVariables() {
        return `
    ${this.getVarModifier('alpha')} float alpha;
    `;
    }
    getUniformAttrs() {
        return [{ name: 'alpha', type: 'float' }];
    }
    getCompilationInfo(input) {
        const info = super.getCompilationInfo(input);
        return Object.assign(Object.assign({}, info), { alpha: input.alpha });
    }
    getInputInfoString(input) {
        return `${super.getInputInfoString(input)}-${input.alpha}`;
    }
}
//# sourceMappingURL=multiply.js.map