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