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_add import Add 

13from .op_and import And 

14from .op_argmax import ArgMax 

15from .op_argmin import ArgMin 

16from .op_array_feature_extractor import ArrayFeatureExtractor 

17from .op_asin import Asin 

18from .op_asinh import Asinh 

19from .op_atan import Atan 

20from .op_atanh import Atanh 

21from .op_average_pool import AveragePool 

22from .op_batch_normalization import BatchNormalization, BatchNormalization_14 

23from .op_binarizer import Binarizer 

24from .op_bitshift import BitShift 

25from .op_broadcast_gradient_args import BroadcastGradientArgs 

26from .op_cast import Cast, CastLike 

27from .op_cdist import CDist 

28from .op_ceil import Ceil 

29from .op_celu import Celu 

30from .op_clip import Clip_6, Clip_11, Clip 

31from .op_category_mapper import CategoryMapper 

32from .op_complex_abs import ComplexAbs 

33from .op_compress import Compress 

34from .op_concat import Concat 

35from .op_concat_from_sequence import ConcatFromSequence 

36from .op_conv import Conv 

37from .op_conv_transpose import ConvTranspose 

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

39from .op_constant_of_shape import ConstantOfShape 

40from .op_cos import Cos 

41from .op_cosh import Cosh 

42from .op_cum_sum import CumSum 

43from .op_debug import DEBUG 

44from .op_det import Det 

45from .op_dequantize_linear import DequantizeLinear 

46from .op_dict_vectorizer import DictVectorizer 

47from .op_div import Div 

48from .op_dropout import Dropout, Dropout_7, Dropout_12 

49from .op_einsum import Einsum 

50from .op_elu import Elu 

51from .op_equal import Equal 

52from .op_erf import Erf 

53from .op_exp import Exp 

54from .op_expand import Expand, Expand_13 

55from .op_eyelike import EyeLike 

56from .op_feature_vectorizer import FeatureVectorizer 

57from .op_fft import FFT 

58from .op_fft2d import FFT2D 

59from .op_flatten import Flatten 

60from .op_fused_matmul import FusedMatMul 

61from .op_gather import Gather 

62from .op_gather_elements import GatherElements 

63from .op_gemm import Gemm 

64from .op_global_average_pool import GlobalAveragePool 

65from .op_greater import Greater, GreaterOrEqual 

66from .op_hardmax import Hardmax 

67from .op_hard_sigmoid import HardSigmoid 

68from .op_floor import Floor 

69from .op_identity import Identity 

70from .op_if import If 

71from .op_isinf import IsInf 

72from .op_imputer import Imputer 

73from .op_isnan import IsNaN 

74from .op_label_encoder import LabelEncoder 

75from .op_leaky_relu import LeakyRelu 

76from .op_less import Less, LessOrEqual 

77from .op_linear_classifier import LinearClassifier 

78from .op_linear_regressor import LinearRegressor 

79from .op_log import Log 

80from .op_log_softmax import LogSoftmax 

81from .op_loop import Loop 

82from .op_lp_normalization import LpNormalization 

83from .op_matmul import MatMul 

84from .op_max import Max 

85from .op_max_pool import MaxPool 

86from .op_mean import Mean 

87from .op_min import Min 

88from .op_mod import Mod 

89from .op_mul import Mul 

90from .op_neg import Neg 

91from .op_negative_log_likelihood_loss import NegativeLogLikelihoodLoss 

92from .op_normalizer import Normalizer 

93from .op_not import Not 

94from .op_one_hot_encoder import OneHotEncoder 

95from .op_or import Or 

96from .op_pad import Pad 

97from .op_pow import Pow 

98from .op_prelu import PRelu 

99from .op_quantize_linear import QuantizeLinear 

100from .op_qlinear_conv import QLinearConv 

101from .op_random import ( 

102 Bernoulli, RandomNormal, RandomUniform, 

103 RandomUniformLike, RandomNormalLike) 

104from .op_range import Range 

105from .op_reciprocal import Reciprocal 

106from .op_reduce_log_sum import ReduceLogSum 

107from .op_reduce_log_sum_exp import ReduceLogSumExp 

108from .op_reduce_l1 import ReduceL1 

109from .op_reduce_l2 import ReduceL2 

110from .op_reduce_min import ReduceMin 

111from .op_reduce_max import ReduceMax 

112from .op_reduce_mean import ReduceMean 

113from .op_reduce_prod import ReduceProd 

114from .op_reduce_sum import ( 

115 ReduceSum_1, ReduceSum_11, ReduceSum_13, ReduceSum) 

116from .op_reduce_sum_square import ReduceSumSquare 

117from .op_relu import Relu 

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

119from .op_rfft import RFFT 

120from .op_round import Round 

121from .op_rnn import RNN 

122from .op_scaler import Scaler 

123from .op_scan import Scan 

124from .op_scatter_elements import ScatterElements 

125from .op_softmax_cross_entropy_loss import SoftmaxCrossEntropyLoss 

126from .op_selu import Selu 

127from .op_sequence_at import SequenceAt 

128from .op_sequence_construct import SequenceConstruct 

129from .op_sequence_insert import SequenceInsert 

130from .op_shape import Shape 

131from .op_sigmoid import Sigmoid 

132from .op_sign import Sign 

133from .op_sin import Sin 

134from .op_sinh import Sinh 

135from .op_size import Size 

136from .op_slice import Slice, Slice_1, Slice_10 

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

138from .op_softmax import Softmax, SoftmaxGrad, SoftmaxGrad_13 

139from .op_solve import Solve 

140from .op_sqrt import Sqrt 

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

142from .op_string_normalizer import StringNormalizer 

143from .op_sub import Sub 

144from .op_sum import Sum 

145from .op_svm_classifier import SVMClassifier, SVMClassifierDouble 

146from .op_svm_regressor import SVMRegressor, SVMRegressorDouble 

147from .op_tan import Tan 

148from .op_tanh import Tanh 

149from .op_tfidfvectorizer import TfIdfVectorizer 

150from .op_tokenizer import Tokenizer 

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

152from .op_transpose import Transpose 

153from .op_tree_ensemble_classifier import ( 

154 TreeEnsembleClassifierDouble, 

155 TreeEnsembleClassifier_1, TreeEnsembleClassifier_3, TreeEnsembleClassifier) 

156from .op_tree_ensemble_regressor import ( 

157 TreeEnsembleRegressorDouble, 

158 TreeEnsembleRegressor_1, TreeEnsembleRegressor_3, TreeEnsembleRegressor) 

159from .op_trilu import Trilu 

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

161from .op_where import Where 

162from .op_xor import Xor 

163from .op_yield_op import YieldOp 

164from .op_zipmap import ZipMap 

165 

166 

167from ..doc.doc_helper import get_rst_doc 

168_op_list = [] 

169clo = locals().copy() 

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

171 if "_" in name: 

172 continue 

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

174 continue # pragma: no cover 

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

176 cl.__doc__ = get_rst_doc(cl.__name__) 

177 _op_list.append(cl)