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 .hadamard_transform import hadamard_transform 

26from .hstack import hstack 

27from .index import index 

28from .index_add import index_add 

29from .index_select import index_select 

30from .isin import isin 

31from .linspace import linspace 

32from .log_softmax import log_softmax, log_softmax_backward 

33from .masked_fill import masked_fill, masked_fill_ 

34from .masked_select import masked_select 

35from .matmul_bf16 import matmul_bf16 

36from .matmul_int8 import matmul_int8 

37from .max import max, max_dim 

38from .mean import mean, mean_dim 

39from .min import min, min_dim 

40from .mm import mm 

41from .multinomial import multinomial 

42from .ones import ones 

43from .ones_like import ones_like 

44from .outer import outer 

45from .polar import polar 

46from .pow import ( 

47 pow_scalar, 

48 pow_tensor_scalar, 

49 pow_tensor_scalar_, 

50 pow_tensor_tensor, 

51 pow_tensor_tensor_, 

52) 

53from .randperm import randperm 

54from .repeat_interleave import repeat_interleave_self_int 

55from .resolve_neg import resolve_neg 

56from .rms_norm import rms_norm 

57from .select_scatter import select_scatter 

58from .slice_scatter import slice_scatter 

59from .softmax import softmax, softmax_backward 

60from .sort import sort 

61from .stack import stack 

62from .threshold import threshold, threshold_backward 

63from .triu import triu 

64from .unique import _unique2 

65from .upsample_nearest2d import upsample_nearest2d 

66from .var_mean import var_mean 

67from .vector_norm import vector_norm 

68from .vstack import vstack 

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

70from .zeros import zeros 

71from .zeros_like import zeros_like 

72 

73__all__ = [ 

74 "_unique2", 

75 "addmm", 

76 "all", 

77 "all_dim", 

78 "all_dims", 

79 "amax", 

80 "angle", 

81 "any", 

82 "any_dim", 

83 "any_dims", 

84 "arange", 

85 "argmax", 

86 "argmin", 

87 "bmm", 

88 "cat", 

89 "count_nonzero", 

90 "cumsum", 

91 "diag", 

92 "diag_embed", 

93 "diagonal_backward", 

94 "dot", 

95 "embedding", 

96 "exponential_", 

97 "fill_scalar", 

98 "fill_scalar_", 

99 "fill_tensor", 

100 "fill_tensor_", 

101 "flip", 

102 "full", 

103 "full_like", 

104 "gather", 

105 "group_norm", 

106 "group_norm_backward", 

107 "hstack", 

108 "hadamard_transform", 

109 "index", 

110 "index_add", 

111 "index_select", 

112 "isin", 

113 "linspace", 

114 "log_softmax", 

115 "log_softmax_backward", 

116 "masked_fill", 

117 "masked_fill_", 

118 "masked_select", 

119 "max", 

120 "max_dim", 

121 "mean", 

122 "mean_dim", 

123 "min", 

124 "min_dim", 

125 "mm", 

126 "matmul_bf16", 

127 "matmul_int8", 

128 "multinomial", 

129 "normed_cumsum", 

130 "ones", 

131 "ones_like", 

132 "outer", 

133 "polar", 

134 "pow_scalar", 

135 "pow_tensor_scalar", 

136 "pow_tensor_scalar_", 

137 "pow_tensor_tensor", 

138 "pow_tensor_tensor_", 

139 "randperm", 

140 "repeat_interleave_self_int", 

141 "resolve_neg", 

142 "rms_norm", 

143 "select_scatter", 

144 "slice_scatter", 

145 "softmax", 

146 "softmax_backward", 

147 "sort", 

148 "stack", 

149 "threshold", 

150 "threshold_backward", 

151 "triu", 

152 "upsample_nearest2d", 

153 "var_mean", 

154 "vector_norm", 

155 "vstack", 

156 "where_self", 

157 "where_self_out", 

158 "where_scalar_other", 

159 "where_scalar_self", 

160 "zeros", 

161 "zeros_like", 

162]