22 lines
941 B
Python
22 lines
941 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .adaptive_avg_pool import (adaptive_avg_pool1d__default,
|
|
adaptive_avg_pool2d__default,
|
|
adaptive_avg_pool2d__ncnn,
|
|
adaptive_avg_pool3d__default)
|
|
from .gelu import gelu__ncnn
|
|
from .grid_sampler import grid_sampler__default
|
|
from .hardsigmoid import hardsigmoid__default
|
|
from .instance_norm import instance_norm__tensorrt
|
|
from .layer_norm import layer_norm__ncnn
|
|
from .linear import linear__ncnn
|
|
from .lstm import generic_rnn__ncnn
|
|
from .squeeze import squeeze__default
|
|
|
|
__all__ = [
|
|
'adaptive_avg_pool1d__default', 'adaptive_avg_pool2d__default',
|
|
'adaptive_avg_pool3d__default', 'grid_sampler__default',
|
|
'hardsigmoid__default', 'instance_norm__tensorrt', 'generic_rnn__ncnn',
|
|
'squeeze__default', 'adaptive_avg_pool2d__ncnn', 'gelu__ncnn',
|
|
'layer_norm__ncnn', 'linear__ncnn'
|
|
]
|