diff --git a/docs/source/ops.rst b/docs/source/ops.rst index ec87d02556e..675aac5bc59 100644 --- a/docs/source/ops.rst +++ b/docs/source/ops.rst @@ -11,7 +11,15 @@ torchvision.ops .. autofunction:: nms .. autofunction:: roi_align +.. autofunction:: ps_roi_align .. autofunction:: roi_pool +.. autofunction:: ps_roi_pool +.. autofunction:: deform_conv2d .. autoclass:: RoIAlign +.. autoclass:: PSRoIAlign .. autoclass:: RoIPool +.. autoclass:: PSRoIPool +.. autoclass:: DeformConv2d +.. autoclass:: MultiScaleRoIAlign +.. autoclass:: FeaturePyramidNetwork