@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
37 lines (36 loc) • 1.39 kB
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input, Operation } from '../operation';
export interface ClipBackwardInfo {
shapeX?: readonly number[];
widthX?: number;
heightX?: number;
shapeGrad?: readonly number[];
widthGrad?: number;
heightGrad?: number;
shapeOutput?: readonly number[];
widthOutput?: number;
heightOutput?: number;
minVal?: number;
maxVal?: number;
doMin?: number;
doMax?: number;
}
export interface ClipBackwardInput {
input: GPUTensorI;
grad: GPUTensorI;
minVal?: number;
maxVal?: number;
}
export declare class ClipBackwardOperation<GPUTensor extends GPUTensorI> extends Operation<GPUTensor, ClipBackwardInfo, ClipBackwardInput> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
getFragmentShader(info: ClipBackwardInfo): string;
getTextureNames(): string[];
getVariables(): string;
getUniformAttrs(): Input[];
calc(input: ClipBackwardInput): GPUTensor;
getOutputShape(input: ClipBackwardInput): readonly number[];
compile(info: ClipBackwardInfo): void;
getCompilationInfo(input: ClipBackwardInput): ClipBackwardInfo;
getInputInfoString(input: ClipBackwardInput): string;
}