@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 712 B
TypeScript
import { GPGPUContext } from './gpgpu_context';
export declare enum MatrixOrientation {
REGULAR = 0,
TRANSPOSED = 1,
}
export declare function getFragmentShaderSource(sharedDimension: number, aOrientation: MatrixOrientation, bOrientation: MatrixOrientation): string;
export declare function multiplyMatrixPacked(gpgpu: GPGPUContext, multiplyProgram: WebGLProgram, a: WebGLTexture, b: WebGLTexture, result: WebGLTexture, resultShapeRowCol: [number, number]): void;
export declare function uploadMultiplyMatrixPackedDownload(a: Float32Array, aShapeRowCol: [number, number], b: Float32Array, bShapeRowCol: [number, number], aOrientation?: MatrixOrientation, bOrientation?: MatrixOrientation): Float32Array;