下载
中文
注册

ScopeROIAlignPass

功能说明

将tf.AvgPool和tf.image.CropAndResize这两个Scope融合为ROIAlign算子,不包含Merge算子。

Scope详情

融合后的算子原型

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

融合对应关系

crop_and_resize/Shape/Switch输入作为融合后的第一个输入features。

Shape输入作为融合后的第二个输入rois。

Avgpool输出作为融合后的输出y。

Shape的输出作为StridedSlice的输入。

StridedSlice的输出作为zeros的输入。

crop_and_resize/Shape/Switch的输出作为crop_and_resize/Shape/Shape和crop_and_resize/transpose的输入。

crop_and_resize/Shape/Shape的输出作为crop_and_resize/strided_slice的输入。

crop_and_resize/strided_slice的输出作为crop_and_resize/transform_fpcoor_for_tf的输入。

crop_and_resize/transform_fpcoor_for_tf和crop_and_resize/transpose的输出作为CropAndResize的输入。

crop_and_resize/CropAndResize的输出作为crop_and_resize/transpose_1的输入。

crop_and_resize/transpose_1的输出作为AvgPool的输入。

适用网络

2D-H1

融合规则类型

定制化融合规则