提交 d4e2d272 编写于 作者: M Megvii Engine Team

feat(mge): do not export F.nn.conv_bias_activation

GitOrigin-RevId: e536f87fad1f16dd3f3501f65210d2573a939434
上级 e64760fe
......@@ -10,7 +10,6 @@
from .elemwise import *
from .math import *
from .nn import *
from .quantized import conv_bias_activation
from .tensor import *
from .utils import *
......
......@@ -1526,4 +1526,5 @@ def nms(boxes: Tensor, scores: Tensor, iou_thresh: float) -> Tensor:
from .loss import *
from .loss import * # isort:skip
from .quantized import conv_bias_activation # isort:skip
......@@ -11,7 +11,7 @@ import numpy as np
from ... import module as Float
from ...core.tensor import dtype
from ...functional import conv_bias_activation
from ...functional.nn import conv_bias_activation
from ...tensor import Parameter
from ..qat import conv as QAT
from .module import QuantizedModule
......
......@@ -438,7 +438,7 @@ def test_conv_bias():
inp = convert_to_nchw4(inp)
w = convert_to_nchw4(w)
b = convert_to_nchw4(b)
return F.conv_bias_activation(
return F.nn.conv_bias_activation(
inp,
w,
b,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册