Browse Source

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

GitOrigin-RevId: e536f87fad
release-1.1
Megvii Engine Team 4 years ago
parent
commit
d4e2d27213
4 changed files with 4 additions and 4 deletions
  1. +0
    -1
      imperative/python/megengine/functional/__init__.py
  2. +2
    -1
      imperative/python/megengine/functional/nn.py
  3. +1
    -1
      imperative/python/megengine/module/quantized/conv.py
  4. +1
    -1
      imperative/python/test/unit/functional/test_functional.py

+ 0
- 1
imperative/python/megengine/functional/__init__.py View File

@@ -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 *



+ 2
- 1
imperative/python/megengine/functional/nn.py View File

@@ -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

+ 1
- 1
imperative/python/megengine/module/quantized/conv.py View File

@@ -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


+ 1
- 1
imperative/python/test/unit/functional/test_functional.py View File

@@ -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,


Loading…
Cancel
Save