Coverage for mlprodict/onnxrt/ops_cpu/_op_list.py: 100%

1# -*- encoding: utf-8 -*- 

2# pylint: disable=W0611 

3""" 

4@file 

5@brief Imports runtime operators. 

6""" 

7 

8from ._op import OpRun 

9from .op_abs import Abs 

10from .op_acos import Acos 

11from .op_acosh import Acosh 

12from .op_adagrad import Adagrad 

13from .op_adam import Adam 

14from .op_add import Add 

15from .op_and import And 

16from .op_argmax import ArgMax 

17from .op_argmin import ArgMin 

18from .op_array_feature_extractor import ArrayFeatureExtractor 

19from .op_asin import Asin 

20from .op_asinh import Asinh 

21from .op_atan import Atan 

22from .op_atanh import Atanh 

23from .op_average_pool import AveragePool 

24from .op_batch_normalization import BatchNormalization, BatchNormalization_14 

25from .op_binarizer import Binarizer 

26from .op_bitshift import BitShift 

27from .op_broadcast_gradient_args import BroadcastGradientArgs 

28from .op_cast import Cast, CastLike 

29from .op_cdist import CDist 

30from .op_ceil import Ceil 

31from .op_celu import Celu 

32from .op_clip import Clip_6, Clip_11, Clip 

33from .op_category_mapper import CategoryMapper 

34from .op_complex_abs import ComplexAbs 

35from .op_compress import Compress 

36from .op_concat import Concat 

37from .op_concat_from_sequence import ConcatFromSequence 

38from .op_conv import Conv 

39from .op_conv_transpose import ConvTranspose 

40from .op_constant import Constant, Constant_12, Constant_11, Constant_9 

41from .op_constant_of_shape import ConstantOfShape 

42from .op_cos import Cos 

43from .op_cosh import Cosh 

44from .op_cum_sum import CumSum 

45from .op_debug import DEBUG 

46from .op_det import Det 

47from .op_depth_to_space import DepthToSpace, SpaceToDepth 

48from .op_dequantize_linear import DequantizeLinear 

49from .op_dft import DFT 

50from .op_dict_vectorizer import DictVectorizer 

51from .op_div import Div 

52from .op_dropout import Dropout, Dropout_7, Dropout_12 

53from .op_einsum import Einsum 

54from .op_elu import Elu 

55from .op_equal import Equal 

56from .op_erf import Erf 

57from .op_exp import Exp 

58from .op_expand import Expand, Expand_13 

59from .op_expression import Expression 

60from .op_eyelike import EyeLike 

61from .op_feature_vectorizer import FeatureVectorizer 

62from .op_fft import FFT 

63from .op_fft2d import FFT2D 

64from .op_flatten import Flatten 

65from .op_fused_matmul import FusedMatMul 

66from .op_gather import Gather 

67from .op_gathernd import GatherND 

68from .op_gather_elements import GatherElements 

69from .op_gemm import Gemm 

70from .op_global_average_pool import GlobalAveragePool, GlobalMaxPool 

71from .op_greater import Greater, GreaterOrEqual 

72from .op_grid_sample import GridSample 

73from .op_gru import GRU 

74from .op_hardmax import Hardmax 

75from .op_hard_sigmoid import HardSigmoid 

76from .op_floor import Floor 

77from .op_identity import Identity 

78from .op_if import If 

79from .op_imputer import Imputer 

80from .op_inverse import Inverse 

81from .op_isinf import IsInf 

82from .op_isnan import IsNaN 

83from .op_label_encoder import LabelEncoder 

84from .op_layer_normalization import LayerNormalization 

85from .op_leaky_relu import LeakyRelu 

86from .op_less import Less, LessOrEqual 

87from .op_linear_classifier import LinearClassifier 

88from .op_linear_regressor import LinearRegressor 

89from .op_log import Log 

90from .op_log_softmax import LogSoftmax 

91from .op_loop import Loop 

92from .op_lp_normalization import LpNormalization 

93from .op_lrn import LRN 

94from .op_lstm import LSTM 

95from .op_matmul import MatMul 

96from .op_max import Max 

97from .op_max_pool import MaxPool 

98from .op_mean import Mean 

99from .op_min import Min 

100from .op_mod import Mod 

101from .op_momentum import Momentum 

102from .op_mul import Mul 

103from .op_murmurhash3 import MurmurHash3 

104from .op_neg import Neg 

105from .op_negative_log_likelihood_loss import NegativeLogLikelihoodLoss 

106from .op_normalizer import Normalizer 

107from .op_non_max_suppression import NonMaxSuppression 

108from .op_non_zero import NonZero 

109from .op_not import Not 

110from .op_one_hot import OneHot 

111from .op_one_hot_encoder import OneHotEncoder 

112from .op_optional import OptionalGetElement, OptionalHasElement 

113from .op_or import Or 

114from .op_pad import Pad 

115from .op_pow import Pow 

116from .op_prelu import PRelu 

117from .op_quantize_linear import QuantizeLinear, DynamicQuantizeLinear 

118from .op_qlinear_conv import QLinearConv 

119from .op_random import ( 

120 Bernoulli, RandomNormal, RandomUniform, 

121 RandomUniformLike, RandomNormalLike) 

122from .op_range import Range 

123from .op_reciprocal import Reciprocal 

124from .op_reduce_log_sum import ( 

125 ReduceLogSum, ReduceLogSum_1, ReduceLogSum_18) 

126from .op_reduce_log_sum_exp import ( 

127 ReduceLogSumExp, ReduceLogSumExp_1, ReduceLogSumExp_18) 

128from .op_reduce_l1 import ReduceL1, ReduceL1_1, ReduceL1_18 

129from .op_reduce_l2 import ReduceL2, ReduceL2_1, ReduceL2_18 

130from .op_reduce_min import ReduceMin, ReduceMin_1, ReduceMin_18 

131from .op_reduce_max import ReduceMax, ReduceMax_1, ReduceMax_18 

132from .op_reduce_mean import ReduceMean_1, ReduceMean_18, ReduceMean 

133from .op_reduce_prod import ReduceProd, ReduceProd_1, ReduceProd_18 

134from .op_reduce_sum import ( 

135 ReduceSum_1, ReduceSum_11, ReduceSum_13, ReduceSum) 

136from .op_reduce_sum_square import ( 

137 ReduceSumSquare, ReduceSumSquare_1, ReduceSumSquare_18) 

138from .op_relu import Relu, ThresholdedRelu 

139from .op_reshape import Reshape, Reshape_5, Reshape_13, Reshape_14 

140from .op_resize import Resize 

141from .op_rfft import RFFT 

142from .op_roi_align import RoiAlign 

143from .op_round import Round 

144from .op_rnn import RNN 

145from .op_scaler import Scaler 

146from .op_scan import Scan 

147from .op_scatter_elements import ScatterElements 

148from .op_scatternd import ScatterND 

149from .op_softmax_cross_entropy_loss import SoftmaxCrossEntropyLoss 

150from .op_selu import Selu 

151from .op_sequence_at import SequenceAt 

152from .op_sequence_construct import SequenceConstruct 

153from .op_sequence_empty import SequenceEmpty 

154from .op_sequence_insert import SequenceInsert 

155from .op_shape import Shape 

156from .op_shrink import Shrink 

157from .op_sigmoid import Sigmoid 

158from .op_sign import Sign 

159from .op_sin import Sin 

160from .op_sinh import Sinh 

161from .op_size import Size 

162from .op_slice import Slice, Slice_1, Slice_10 

163from .op_split import Split, Split_2, Split_11, Split_13 

164from .op_softmax import ( 

165 Softmax, Softmax_1, Softmax_13, SoftmaxGrad, SoftmaxGrad_13) 

166from .op_softplus import Softplus 

167from .op_softsign import Softsign 

168from .op_solve import Solve 

169from .op_sqrt import Sqrt 

170from .op_squeeze import Squeeze, Squeeze_1, Squeeze_11, Squeeze_13 

171from .op_stft import STFT 

172from .op_string_normalizer import StringNormalizer 

173from .op_sub import Sub 

174from .op_sum import Sum 

175from .op_svm_classifier import SVMClassifier, SVMClassifierDouble 

176from .op_svm_regressor import SVMRegressor, SVMRegressorDouble 

177from .op_tan import Tan 

178from .op_tanh import Tanh 

179from .op_tfidfvectorizer import TfIdfVectorizer 

180from .op_tokenizer import Tokenizer 

181from .op_topk import TopK_10, TopK_11, TopK_1, TopK 

182from .op_transpose import Transpose 

183from .op_tree_ensemble_classifier import ( 

184 TreeEnsembleClassifierDouble, 

185 TreeEnsembleClassifier_1, TreeEnsembleClassifier_3, TreeEnsembleClassifier) 

186from .op_tree_ensemble_regressor import ( 

187 TreeEnsembleRegressorDouble, 

188 TreeEnsembleRegressor_1, TreeEnsembleRegressor_3, TreeEnsembleRegressor) 

189from .op_trilu import Trilu 

190from .op_unique import Unique 

191from .op_unsqueeze import Unsqueeze, Unsqueeze_1, Unsqueeze_11, Unsqueeze_13 

192from .op_where import Where 

193from .op_window import BlackmanWindow, HannWindow, HammingWindow 

194from .op_xor import Xor 

195from .op_yield_op import YieldOp 

196from .op_zipmap import ZipMap 

197from ..doc.doc_helper import get_rst_doc 

198 

199_op_list = [] 

200clo = locals().copy() 

201for name, cl in clo.items(): 

202 if "_" in name: 

203 continue 

204 if name in {'cl', 'clo', 'name'}: 

205 continue # pragma: no cover 

206 if not cl.__doc__ and issubclass(cl, OpRun): 

207 cl.__doc__ = get_rst_doc(cl.__name__) 

208 _op_list.append(cl)