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

PyTorch like deep learning inferrence library

import { DTypeGpu, GPUTensorI } from '../../tensor/gpu/interface';
import { Operation } from './operation';
interface OpInfo<Op> {
    operation?: Op;
    numCalls: number;
    infoString: string;
}
export declare class Dispatcher<GPUTensor extends GPUTensorI, Info, Input, Op extends Operation<GPUTensor, Info, Input>> {
    private getOp;
    private minCallsToCompile;
    private opDict;
    constructor(getOp: (dtype: DTypeGpu) => Op, minCallsToCompile?: number);
    getDefault(dtype: DTypeGpu): OpInfo<Op>;
    calc(input: Input, dtype: DTypeGpu): GPUTensor;
}
export {};