# 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

_overlap_3d = torch.ops.nodulenet.overlap_3d

overlap_3d = amp.float_function(_overlap_3d)