7 lines
321 B
TypeScript
7 lines
321 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
|
|
import { PredictionLayerParams } from './types';
|
|
export declare function predictionLayer(x: tf.Tensor4D, conv11: tf.Tensor4D, params: PredictionLayerParams): {
|
|
boxPredictions: tf.Tensor4D;
|
|
classPredictions: tf.Tensor4D;
|
|
};
|
|
//# sourceMappingURL=predictionLayer.d.ts.map
|