# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
import torch

# Only valid with fp32 inputs - give AMP the hint
# from ..utils import amp
import apex.amp as amp

_nms_3d = torch.ops.nodulenet.nms_3d

nms_3d = amp.float_function(_nms_3d)