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 "addmm", 

72 "all", 

73 "all_dim", 

74 "all_dims", 

75 "amax", 

76 "argmax", 

77 "bmm", 

78 "fill_scalar", 

79 "fill_scalar_", 

80 "fill_tensor", 

81 "fill_tensor_", 

82 "max", 

83 "max_dim", 

84 "min", 

85 "min_dim", 

86 "mm", 

87 "pow_scalar", 

88 "pow_tensor_scalar", 

89 "pow_tensor_scalar_", 

90 "pow_tensor_tensor", 

91 "pow_tensor_tensor_", 

92 "triu", 

93 "resolve_neg", 

94 "rms_norm", 

95 "cat", 

96 "count_nonzero", 

97 "cumsum", 

98 "normed_cumsum", 

99 "diag", 

100 "diagonal_backward", 

101 "diag_embed", 

102 "dot", 

103 "embedding", 

104 "exponential_", 

105 "flip", 

106 "full", 

107 "full_like", 

108 "masked_fill", 

109 "masked_fill_", 

110 "masked_select", 

111 "mean", 

112 "mean_dim", 

113 "where_self_out", 

114 "where_self", 

115 "where_scalar_self", 

116 "where_scalar_other", 

117 "index", 

118 "index_select", 

119 "isin", 

120 "gather", 

121 "group_norm", 

122 "group_norm_backward", 

123 "hstack", 

124 "polar", 

125 "repeat_interleave_self_int", 

126 "select_scatter", 

127 "slice_scatter", 

128 "softmax", 

129 "softmax_backward", 

130 "sort", 

131 "stack", 

132 "linspace", 

133 "log_softmax", 

134 "log_softmax_backward", 

135 "zeros", 

136 "vector_norm", 

137 "outer", 

138 "arange", 

139 "threshold", 

140 "threshold_backward", 

141 "zeros_like", 

142 "ones", 

143 "ones_like", 

144 "argmin", 

145 "var_mean", 

146 "vstack", 

147 "any", 

148 "any_dims", 

149 "any_dim", 

150 "angle", 

151 "multinomial", 

152 "index_add", 

153 "_unique2", 

154 "upsample_nearest2d", 

155 "randperm", 

156]