Coverage for src/flag_gems/runtime/backend/_ascend/ops/__init__.py: 0%

1from .addmm import addmm 

2from .all import all, all_dim, all_dims 

3from .amax import amax 

4from .angle import angle 

5from .any import any, any_dim, any_dims 

6from .arange import arange, arange_start 

7from .argmax import argmax 

8from .argmin import argmin 

9from .attention import ( 

10 ScaleDotProductAttention, 

11 flash_attention_forward, 

12 flash_attn_varlen_func, 

13 scaled_dot_product_attention, 

14 scaled_dot_product_attention_backward, 

15 scaled_dot_product_attention_forward, 

16) 

17from .baddbmm import baddbmm 

18from .bmm import bmm 

19from .cat import cat, cat_out 

20from .count_nonzero import count_nonzero 

21from .cummax import cummax 

22from .cummin import cummin 

23from .cumsum import cumsum, normed_cumsum 

24from .diag import diag 

25from .diag_embed import diag_embed 

26from .diagonal import diagonal_backward 

27from .dot import dot 

28from .embedding import embedding 

29from .exponential_ import exponential_ 

30from .fill import fill_scalar, fill_scalar_, fill_tensor, fill_tensor_ 

31from .flip import flip 

32from .full import full 

33from .full_like import full_like 

34from .gather import gather, gather_backward 

35from .groupnorm import group_norm, group_norm_backward 

36from .hadamard_transform import hadamard_transform 

37from .hstack import hstack 

38from .index import index 

39from .index_add import index_add, index_add_ 

40from .index_select import index_select 

41from .isin import isin 

42from .linspace import linspace 

43from .log_softmax import log_softmax, log_softmax_backward, log_softmax_out 

44from .masked_fill import masked_fill, masked_fill_ 

45from .masked_scatter import masked_scatter, masked_scatter_ 

46from .masked_select import masked_select 

47from .matmul_bf16 import matmul_bf16 

48from .matmul_int8 import matmul_int8 

49from .max import max, max_dim 

50from .mean import mean, mean_dim 

51from .min import min, min_dim 

52from .mm import mm, mm_out 

53from .multinomial import multinomial 

54from .ones import ones 

55from .ones_like import ones_like 

56from .outer import outer 

57from .polar import polar 

58from .pow import ( 

59 pow_scalar, 

60 pow_tensor_scalar, 

61 pow_tensor_scalar_, 

62 pow_tensor_tensor, 

63 pow_tensor_tensor_, 

64) 

65from .randperm import randperm 

66from .repeat_interleave import repeat_interleave_self_int 

67from .resolve_neg import resolve_neg 

68from .rms_norm import rms_norm 

69from .scatter import scatter, scatter_ 

70from .scatter_add_ import scatter_add_ 

71from .select_backward import select_backward 

72from .select_scatter import select_scatter 

73from .slice_scatter import slice_scatter 

74from .softmax import softmax, softmax_backward, softmax_backward_out, softmax_out 

75from .sort import sort 

76from .stack import stack 

77from .threshold import threshold, threshold_backward 

78from .triu import triu 

79from .unique import _unique2 

80from .upsample_bicubic2d_aa import _upsample_bicubic2d_aa 

81from .upsample_nearest2d import upsample_nearest2d 

82from .var_mean import var_mean 

83from .vector_norm import vector_norm 

84from .vstack import vstack 

85from .where import where_scalar_other, where_scalar_self, where_self, where_self_out 

86from .zeros import zeros 

87from .zeros_like import zeros_like 

88 

89__all__ = [ 

90 "_unique2", 

91 "_upsample_bicubic2d_aa", 

92 "addmm", 

93 "all", 

94 "all_dim", 

95 "all_dims", 

96 "amax", 

97 "angle", 

98 "any", 

99 "any_dim", 

100 "any_dims", 

101 "arange", 

102 "arange_start", 

103 "argmax", 

104 "argmin", 

105 "baddbmm", 

106 "bmm", 

107 "cat", 

108 "cat_out", 

109 "count_nonzero", 

110 "cummax", 

111 "cummin", 

112 "cumsum", 

113 "diag", 

114 "diag_embed", 

115 "diagonal_backward", 

116 "dot", 

117 "embedding", 

118 "exponential_", 

119 "fill_scalar", 

120 "fill_scalar_", 

121 "fill_tensor", 

122 "fill_tensor_", 

123 "flash_attention_forward", 

124 "flash_attn_varlen_func", 

125 "flip", 

126 "full", 

127 "full_like", 

128 "gather", 

129 "gather_backward", 

130 "group_norm", 

131 "group_norm_backward", 

132 "hadamard_transform", 

133 "hstack", 

134 "index", 

135 "index_add", 

136 "index_add_", 

137 "index_select", 

138 "isin", 

139 "linspace", 

140 "log_softmax", 

141 "log_softmax_backward", 

142 "log_softmax_out", 

143 "masked_fill", 

144 "masked_fill_", 

145 "masked_scatter", 

146 "masked_scatter_", 

147 "masked_select", 

148 "matmul_bf16", 

149 "matmul_int8", 

150 "max", 

151 "max_dim", 

152 "mean", 

153 "mean_dim", 

154 "min", 

155 "min_dim", 

156 "mm", 

157 "mm_out", 

158 "multinomial", 

159 "normed_cumsum", 

160 "ones", 

161 "ones_like", 

162 "outer", 

163 "polar", 

164 "pow_scalar", 

165 "pow_tensor_scalar", 

166 "pow_tensor_scalar_", 

167 "pow_tensor_tensor", 

168 "pow_tensor_tensor_", 

169 "randperm", 

170 "repeat_interleave_self_int", 

171 "resolve_neg", 

172 "rms_norm", 

173 "scatter", 

174 "scatter_", 

175 "scatter_add_", 

176 "ScaleDotProductAttention", 

177 "scaled_dot_product_attention", 

178 "scaled_dot_product_attention_backward", 

179 "scaled_dot_product_attention_forward", 

180 "select_backward", 

181 "select_scatter", 

182 "slice_scatter", 

183 "softmax", 

184 "softmax_backward", 

185 "softmax_backward_out", 

186 "softmax_out", 

187 "sort", 

188 "stack", 

189 "threshold", 

190 "threshold_backward", 

191 "triu", 

192 "upsample_nearest2d", 

193 "var_mean", 

194 "vector_norm", 

195 "vstack", 

196 "where_scalar_other", 

197 "where_scalar_self", 

198 "where_self", 

199 "where_self_out", 

200 "zeros", 

201 "zeros_like", 

202]