下载
中文
注册

ScopeFastrcnnPredictionsPass

功能说明

将fastrcnn_predictions Scope融合为FastrcnnPredictions算子。Scope内包括2的倍数个tf.TopKV2算子、3的倍数个tf.Where算子、tf. NonMaxSuppressionV2算子、tf.Less算子、tf.LoopCond算子等,不包含ExpandDims、clip_boxes和decode_bbox_target。

Scope详情

融合后的算子原型

FastrcnnPredictions,具体请参见算子加速库接口参考

融合对应关系

fastrcnn_predictions/transpose和fastrcnn_predictions/GatherNd的输入作为融合后的输入rois。

fastrcnn_predictions/strided_slice的输入作为融合后的输入score。

fastrcnn_predictions/TopKV2的输出作为融合的输出sorted_rois。

fastrcnn_predictions/GatherNd的输出作为融合后的输出sorted_scores。

fastrcnn_predictions/Add的输出作为融合后的输出sorted_classes。

fastrcnn_predictions/strided_slice的输出作为fastrcnn_predictions/transpose_1的输入。

fastrcnn_predictions/transpose_1的输出作为fastrcnn_predictions/map和fastrcnn_predictions/boolean_mask的输入。

astrcnn_predictions/map的输出作为fastrcnn_predictions/Where的输入。

fastrcnn_predictions/Where的输出作为fastrcnn_predictions/Gather的输入。

fastrcnn_predictions/boolean_mask的输出作为fastrcnn_predictions/Size和fastrcnn_predictions/TopKV2的输入。

fastrcnn_predictions/Size的输出作为fastrcnn_predictions/Minimum的输入。

fastrcnn_predictions/Minimum的输出作为fastrcnn_predictions/TopKV2的输入。

fastrcnn_predictions/TopKV2作为融合后的输出(sorted_rois)和fastrcnn_predictions/Gather的输入。

fastrcnn_predictions/Gather的输出作为fastrcnn_predictions/filtered_indices的输入。

fastrcnn_predictions/filtered_indices的输出作为fastrcnn_predictions/GatherNd和fastrcnn_predictions/ToFloat的输入。

fastrcnn_predictions/ToFloat的输出作为fastrcnn_predictions/strided_slice_1的输入。

fastrcnn_predictions/strided_slice_1的输出作为fastrcnn_predictions/Add的输入。

适用网络

2D-H1

融合规则类型

定制化融合规则