@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
8 lines (7 loc) • 362 B
TypeScript
import { GPGPUProgram } from './gpgpu_math';
export declare class MatMulProgram implements GPGPUProgram {
variableNames: string[];
outputShape: number[];
userCode: string;
constructor(aShape: [number, number, number], bShape: [number, number, number], transposeA?: boolean, transposeB?: boolean, addBias?: boolean, activation?: string);
}