fd55cf8ea1dd674fbe60ade84709fe5666dd9434,pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py,ThreeInterpolate,backward,#,163

Before Change

            grad_out.contiguous(), idx, weight, m
        )

        return grad_features, None, None


three_interpolate = ThreeInterpolate.apply

After Change

        None
        
        idx, weight, features = ctx.saved_tensors
        m = features.size(2)

        grad_features = _ext.three_interpolate_grad(
            grad_out.contiguous(), idx, weight, m
        )

        return grad_features, torch.zeros_like(idx), torch.zeros_like(weight)


three_interpolate = ThreeInterpolate.apply

Instances