@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
39 lines (38 loc) • 1.49 kB
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input, Operation } from '../operation';
export interface AveragePoolInfo {
shapeX?: readonly number[];
widthX?: number;
heightX?: number;
kernelShape?: readonly number[];
shapeOutput?: readonly number[];
widthOutput?: number;
heightOutput?: number;
pads?: readonly number[];
strides?: readonly number[];
kernelSize?: number;
dataRank?: number;
includePad?: number | boolean;
}
export interface AveragePoolInput {
X: GPUTensorI;
pads: number[];
strides: number[];
kernelShape: number[];
includePad: boolean;
}
export declare class AveragePoolOperation<GPUTensor extends GPUTensorI> extends Operation<GPUTensor, AveragePoolInfo, AveragePoolInput> {
protected maxIterations: number;
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
updateInputIx(): string;
getVariables(): string;
getFragmentShader(info: AveragePoolInfo): string;
getTextureNames(): string[];
getUniformAttrs(): Input[];
calc(input: AveragePoolInput): GPUTensor;
getOutputShape(input: AveragePoolInput): readonly number[];
compile(info: AveragePoolInfo): void;
getCompilationInfo(input: AveragePoolInput): AveragePoolInfo;
getInputInfoString(input: AveragePoolInput): string;
}