@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 335 B
TypeScript
export interface SegOpInfo {
windowSize: number;
batchSize: number;
inSize: number;
numSegments: number;
}
export declare function segOpComputeOptimalWindowSize(inSize: number, numSegments: number): number;
export declare function computeOutShape(aShape: number[], axis: number, numSegments: number): number[];