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

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

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