@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
14 lines (13 loc) • 467 B
TypeScript
export declare type ShapeInfo = {
logicalShape: number[];
texShape: [number, number];
isUniform: boolean;
isPacked: boolean;
flatOffset: number;
};
export declare type InputInfo = {
name: string;
shapeInfo: ShapeInfo;
};
export declare function makeShader(inputsInfo: InputInfo[], outputShape: ShapeInfo, userCode: string, usesPackedTextures: boolean): string;
export declare function getCoordsDataType(rank: number): string;