6 lines
302 B
TypeScript
6 lines
302 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.Tensor<tf.Rank.R4>;
|
|
classPredictions: tf.Tensor<tf.Rank.R4>;
|
|
};
|