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 

7from .argmax import argmax 

8from .argmin import argmin 

9from .bmm import bmm 

10from .cat import cat 

11from .count_nonzero import count_nonzero 

12from .cumsum import cumsum, normed_cumsum 

13from .diag import diag 

14from .diag_embed import diag_embed 

15from .diagonal import diagonal_backward 

16from .dot import dot 

17from .embedding import embedding 

18from .exponential_ import exponential_ 

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

20from .flip import flip 

21from .full import full 

22from .full_like import full_like 

23from .gather import gather 

24from .groupnorm import group_norm, group_norm_backward 

25from .hstack import hstack 

26from .index import index 

27from .index_add import index_add 

28from .index_select import index_select 

29from .isin import isin 

30from .linspace import linspace 

31from .log_softmax import log_softmax, log_softmax_backward 

32from .masked_fill import masked_fill, masked_fill_ 

33from .masked_select import masked_select 

34from .max import max, max_dim 

35from .mean import mean, mean_dim 

36from .min import min, min_dim 

37from .mm import mm 

38from .multinomial import multinomial 

39from .ones import ones 

40from .ones_like import ones_like 

41from .outer import outer 

42from .polar import polar 

43from .pow import ( 

44 pow_scalar, 

45 pow_tensor_scalar, 

46 pow_tensor_scalar_, 

47 pow_tensor_tensor, 

48 pow_tensor_tensor_, 

49) 

50from .randperm import randperm 

51from .repeat_interleave import repeat_interleave_self_int 

52from .resolve_neg import resolve_neg 

53from .rms_norm import rms_norm 

54from .select_scatter import select_scatter 

55from .slice_scatter import slice_scatter 

56from .softmax import softmax, softmax_backward 

57from .sort import sort 

58from .stack import stack 

59from .threshold import threshold, threshold_backward 

60from .triu import triu 

61from .unique import _unique2 

62from .upsample_nearest2d import upsample_nearest2d 

63from .var_mean import var_mean 

64from .vector_norm import vector_norm 

65from .vstack import vstack 

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

67from .zeros import zeros 

68from .zeros_like import zeros_like 

69 

70__all__ = [ 

71 "_unique2", 

72 "addmm", 

73 "all", 

74 "all_dim", 

75 "all_dims", 

76 "amax", 

77 "angle", 

78 "any", 

79 "any_dim", 

80 "any_dims", 

81 "arange", 

82 "argmax", 

83 "argmin", 

84 "bmm", 

85 "cat", 

86 "count_nonzero", 

87 "cumsum", 

88 "diag", 

89 "diag_embed", 

90 "diagonal_backward", 

91 "dot", 

92 "embedding", 

93 "exponential_", 

94 "fill_scalar", 

95 "fill_scalar_", 

96 "fill_tensor", 

97 "fill_tensor_", 

98 "flip", 

99 "full", 

100 "full_like", 

101 "gather", 

102 "group_norm", 

103 "group_norm_backward", 

104 "hstack", 

105 "index", 

106 "index_add", 

107 "index_select", 

108 "isin", 

109 "linspace", 

110 "log_softmax", 

111 "log_softmax_backward", 

112 "masked_fill", 

113 "masked_fill_", 

114 "masked_select", 

115 "max", 

116 "max_dim", 

117 "mean", 

118 "mean_dim", 

119 "min", 

120 "min_dim", 

121 "mm", 

122 "multinomial", 

123 "normed_cumsum", 

124 "ones", 

125 "ones_like", 

126 "outer", 

127 "polar", 

128 "pow_scalar", 

129 "pow_tensor_scalar", 

130 "pow_tensor_scalar_", 

131 "pow_tensor_tensor", 

132 "pow_tensor_tensor_", 

133 "randperm", 

134 "repeat_interleave_self_int", 

135 "resolve_neg", 

136 "rms_norm", 

137 "select_scatter", 

138 "slice_scatter", 

139 "softmax", 

140 "softmax_backward", 

141 "sort", 

142 "stack", 

143 "threshold", 

144 "threshold_backward", 

145 "triu", 

146 "upsample_nearest2d", 

147 "var_mean", 

148 "vector_norm", 

149 "vstack", 

150 "where_self", 

151 "where_self_out", 

152 "where_scalar_other", 

153 "where_scalar_self", 

154 "zeros", 

155 "zeros_like", 

156]