@hoff97/tensor-js/dist/lib/ops/gpu/unary/addMultiplyScalar.d.ts

PyTorch like deep learning inferrence library

import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input } from '../operation';
import { UnaryOperation, UnaryOpInfo, UnaryOpInput } from './unaryOperation';
interface AddMultiplyScalarInfo extends UnaryOpInfo {
    factor?: number;
    add?: number;
}
interface AddMultiplyScalarInput extends UnaryOpInput {
    factor: number;
    add: number;
}
export declare class AddMultiplyScalarOperation<GPUTensor extends GPUTensorI> extends UnaryOperation<GPUTensor, AddMultiplyScalarInfo, AddMultiplyScalarInput> {
    constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
    operation(input: string): string;
    calc(input: AddMultiplyScalarInput): GPUTensor;
    getCompilationInfo(input: AddMultiplyScalarInput): AddMultiplyScalarInfo;
    getInputInfoString(input: AddMultiplyScalarInput): string;
    getVariables(): string;
    getUniformAttrs(): Input[];
}
export {};