forked from pytorch/xla
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathyaml_same_func.txt
330 lines (330 loc) · 4.09 KB
/
yaml_same_func.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
abs
sgn
acos
add.Tensor
add.Scalar
all.dim
any.dim
arange.start_out
argmax
argmin
acosh
asinh
atanh
as_strided
as_strided_
asin
atan
baddbmm
bernoulli
bernoulli_.Tensor
bernoulli_.float
binary_cross_entropy
binary_cross_entropy_backward
binary_cross_entropy_with_logits
bitwise_not.out
logical_not
logical_xor
logical_and
logical_or
bmm
cat
ceil
clamp
clamp.Tensor
clamp_max
clamp_max.Tensor_out
clamp_min
clamp_min.Tensor_out
constant_pad_nd
convolution_overrideable
convolution_backward_overrideable
_copy_from
_copy_from_and_resize
cos
cosh
cumprod
cumsum
diagonal
div.Tensor
div.Tensor_mode
div.Scalar
dot
embedding
embedding_dense_backward
empty.memory_format
resize_
empty_strided
erf
erfc
exp
expm1
expand
eye.out
eye.m_out
fill_.Scalar
fill_.Tensor
floor
frac
index.Tensor
index_copy
index_put_
_index_put_impl_
inverse
isnan
kl_div
kl_div_backward
kthvalue
nan_to_num
linspace
log
log10
log1p
log2
xlogy.Tensor
logdet
_log_softmax
_log_softmax_backward_data
logsumexp
max.dim
max.dim_max
amax
mean
mean.dim
min.dim
min.dim_min
amin
mm
mul.Tensor
mul.Scalar
mv
mv.out
native_batch_norm
native_batch_norm_backward
permute
reciprocal
neg
repeat
round
relu
relu_
prelu
gelu
gelu_backward
hardshrink
hardshrink_backward
rsqrt
select.int
silu.out
silu_backward
mish
sigmoid
sin
sinh
slice.Tensor
slogdet
_softmax
_softmax_backward_data
split.Tensor
split_with_sizes
squeeze
squeeze.dim
squeeze_
squeeze_.dim
stack
sum
sum.dim_IntList
sqrt
std
std.dim
std.correction
std_mean.correction
prod
prod.dim_int
t
t_
tan
tanh
threshold
threshold_backward
transpose.int
transpose_
flip
roll
_trilinear
trunc
_unsafe_view
unsqueeze
unsqueeze_
var.correction
var_mean.correction
where.self
norm.ScalarOpt_dtype
norm.Scalar
norm.ScalarOpt_dim_dtype
norm.ScalarOpt_dim
clone
zero_
sub.Tensor
sub.Scalar
rsub.Tensor
rsub.Scalar
addmm
_to_cpu
unbind.int
_local_scalar_dense
_pack_padded_sequence
masked_fill_.Scalar
masked_fill_.Tensor
masked_scatter_
view
put_
index_add
index_fill_.int_Scalar
index_fill_.int_Tensor
scatter.src
scatter.value
scatter.reduce
scatter.value_reduce
scatter_add
bitwise_and.Scalar
bitwise_and.Tensor
bitwise_or.Tensor_out
bitwise_or.Scalar_out
bitwise_xor.Tensor_out
bitwise_xor.Scalar_out
__lshift__.Scalar
__lshift__.Tensor
__ilshift__.Scalar
__ilshift__.Tensor
__rshift__.Scalar
__rshift__.Tensor
__irshift__.Scalar
__irshift__.Tensor
tril_
triu_
random_.from
random_.to
random_
uniform_
exponential_
diag
cross
triu
tril
trace
ne.Scalar
ne.Tensor
eq.Scalar
eq.Tensor
ge.Scalar
ge.Tensor
le.Scalar
le.Tensor
gt.Scalar
gt.Tensor
lt.Scalar
lt.Tensor
take
index_select
masked_select
nonzero
gather
addcmul
addcdiv
addcdiv_
triangular_solve
symeig
svd
cholesky
qr
erfinv
sign
atan2
lerp.Scalar
lerp.Tensor
fmod.Scalar
fmod.Tensor
remainder.Scalar
remainder.Tensor
min
max
maximum
minimum
sort
sort.stable
topk
all
any
pow.Tensor_Tensor
pow.Scalar
pow.Tensor_Scalar
normal_
normal.Tensor_float
normal.float_Tensor
normal.Tensor_Tensor
alias
_amp_foreach_non_finite_check_and_unscale_
_amp_update_scale_
mse_loss
mse_loss_backward
l1_loss
l1_loss_backward
nll_loss_forward
nll_loss_backward
nll_loss2d_forward
nll_loss2d_backward
smooth_l1_loss
smooth_l1_loss_backward
elu
elu_backward
elu_
hardsigmoid
hardsigmoid_backward
hardtanh
hardtanh_backward
hardswish
hardswish_backward
leaky_relu
leaky_relu_backward
log_sigmoid_forward
log_sigmoid_backward
rrelu_with_noise
rrelu_with_noise_backward
softplus
softplus_backward
softshrink
softshrink_backward
_adaptive_avg_pool2d
_adaptive_avg_pool2d_backward
_adaptive_avg_pool3d
_adaptive_avg_pool3d_backward
adaptive_max_pool2d
adaptive_max_pool2d_backward
avg_pool2d
avg_pool2d_backward
avg_pool3d
avg_pool3d_backward
max_pool2d_with_indices
max_pool2d_with_indices_backward
max_pool3d_with_indices
max_pool3d_with_indices_backward
max_unpool2d
max_unpool2d_backward
max_unpool3d
max_unpool3d_backward
reflection_pad2d
reflection_pad2d_backward
replication_pad1d
replication_pad1d_backward
replication_pad2d
replication_pad2d_backward
upsample_nearest2d.vec
upsample_nearest2d_backward.vec
upsample_bilinear2d
upsample_bilinear2d_backward
upsample_nearest2d
upsample_nearest2d_backward
sigmoid_backward
tanh_backward
ger