@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
33 lines (32 loc) • 1.28 kB
TypeScript
import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input, Operation } from '../operation';
export interface SetValuesInfo {
shapeA?: readonly number[];
widthA?: number;
heightA?: number;
shapeValues?: readonly number[];
widthValues?: number;
heightValues?: number;
shapeOutput?: readonly number[];
widthOutput?: number;
heightOutput?: number;
starts?: number[];
}
export interface SetValuesInput {
A: GPUTensorI;
Values: GPUTensorI;
starts: number[];
}
export declare class SetValuesOperation<GPUTensor extends GPUTensorI> extends Operation<GPUTensor, SetValuesInfo, SetValuesInput> {
constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
getVariables(): string;
getUniformAttrs(): Input[];
getFragmentShader(info: SetValuesInfo): string;
getTextureNames(): string[];
calc(input: SetValuesInput): GPUTensor;
getOutputShape(input: SetValuesInput): readonly number[];
compile(info: SetValuesInfo): void;
getCompilationInfo(input: SetValuesInput): SetValuesInfo;
getInputInfoString(input: SetValuesInput): string;
}