@hoff97/tensor-js/dist/lib/ops/gpu/unary/round.js

PyTorch like deep learning inferrence library

import { UnaryOperation } from './unaryOperation';
export class RoundOperation extends UnaryOperation {
    constructor(tensorConstructor, dtype, allocator) {
        super(tensorConstructor, dtype, allocator);
    }
    operation(input) {
        return `floor(${input}+vec4(0.5))`;
    }
}
//# sourceMappingURL=round.js.map