@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
38 lines (37 loc) • 1.63 kB
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input, Operation } from '../../../ops/gpu/operation';
import { Dispatcher } from '../../../ops/gpu/dispatcher';
export interface UpdateValueInfo {
shapeValue?: readonly number[];
widthValue?: number;
heightValue?: number;
shapeMoments?: readonly number[];
widthMoments?: number;
heightMoments?: number;
shapeOutput?: readonly number[];
widthOutput?: number;
heightOutput?: number;
alpha?: number;
epsilon?: number;
}
export interface UpdateValueInput {
Value: GPUTensorI;
Moments: GPUTensorI;
alpha: number;
epsilon: number;
}
export declare class UpdateValueOperation<GPUTensor extends GPUTensorI> extends Operation<GPUTensor, UpdateValueInfo, UpdateValueInput> {
protected maxIterations: number;
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
getVariables(): string;
getFragmentShader(info: UpdateValueInfo): string;
getTextureNames(): string[];
getUniformAttrs(): Input[];
calc(input: UpdateValueInput): GPUTensor;
getOutputShape(input: UpdateValueInput): readonly number[];
compile(info: UpdateValueInfo): void;
getCompilationInfo(input: UpdateValueInput): UpdateValueInfo;
getInputInfoString(input: UpdateValueInput): string;
}
export declare const defaultUpdateValueD: Dispatcher<GPUTensorI, unknown, unknown, UpdateValueOperation<import("../../../tensor/gpu/tensor").GPUTensor<DTypeGpu>>>;