mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[functorch] updated gen_data script
This commit is contained in:
Horace He
Jon Janzen
@ -79,25 +79,13 @@ def get_ops_for_key(key):
|
||||
batched_registrations = get_ops_for_key('FuncTorchBatched')
|
||||
all_ops = get_ops_for_key(None)
|
||||
|
||||
# Find all occurrences of things inside of STOP_DECOMPOSE(...) using regex
|
||||
# Look in ../functorch/csrc/BatchRulesStopDecomposition.cpp
|
||||
# Example:
|
||||
# STOP_DECOMPOSE(sin); => sin
|
||||
with open('../functorch/csrc/BatchRulesStopDecomposition.cpp') as f:
|
||||
content = f.read()
|
||||
stop_decomposition_regex = re.compile(r'STOP_DECOMPOSE\((.*)\);')
|
||||
stop_decomposition_matches = stop_decomposition_regex.findall(content)
|
||||
stop_decomposition_matches = [m.strip() for m in stop_decomposition_matches]
|
||||
stop_decomposition_ops = set(stop_decomposition_matches)
|
||||
|
||||
composite_ops = get_ops_for_key('CompositeImplicitAutograd')
|
||||
decomposed_ops = composite_ops - stop_decomposition_ops
|
||||
|
||||
|
||||
vmap_ops = (batched_registrations - stop_decomposition_ops) | (composite_ops - stop_decomposition_ops)
|
||||
vmap_ops = batched_registrations
|
||||
noncomposite_ops = all_ops - composite_ops
|
||||
|
||||
ops = yaml.load(open('/home/chilli/fb/pytorch/aten/src/ATen/native/native_functions.yaml', 'r').read())
|
||||
ops = yaml.load(open('/home/chilli/fb/pytorch/aten/src/ATen/native/native_functions.yaml', 'r').read(), Loader=yaml.CLoader)
|
||||
|
||||
annotated_ops = {a.strip(): b.strip() for a,b in list(csv.reader(open('annotated_ops.txt')))}
|
||||
from collections import defaultdict
|
||||
@ -133,8 +121,6 @@ def annotate_ops(ops, is_unique):
|
||||
categorization['inplace'] += 1
|
||||
op['meta'] = 'inplace'
|
||||
continue
|
||||
if 'slow_conv3d_backward.grad_input' in op['full_name']:
|
||||
import pdb; pdb.set_trace()
|
||||
if not is_unique and 'a!' in op['func'].lower():
|
||||
categorization['out'] += 1
|
||||
op['meta'] = 'out'
|
||||
|
||||
@ -42,7 +42,7 @@ _sobol_engine_scramble_,inplace,True,False
|
||||
_sobol_engine_initialize_state_,inplace,True,False
|
||||
_reshape_from_tensor,private,True,False
|
||||
_shape_as_tensor,private,True,False
|
||||
dropout,core composite pointwise,True,True
|
||||
dropout,core composite pointwise,True,False
|
||||
dropout_,inplace,True,False
|
||||
feature_dropout,core composite pointwise,True,False
|
||||
feature_dropout_,inplace,True,False
|
||||
@ -53,7 +53,7 @@ feature_alpha_dropout_,inplace,True,False
|
||||
abs,core primitive pointwise,False,True
|
||||
abs_,inplace,False,True
|
||||
abs.out,out,False,False
|
||||
absolute,core alias,True,False
|
||||
absolute,core alias,True,True
|
||||
absolute_,inplace,True,False
|
||||
absolute.out,out,True,False
|
||||
angle,core composite pointwise,False,True
|
||||
@ -66,22 +66,22 @@ sgn.out,out,False,False
|
||||
real,core complex,True,True
|
||||
imag,core complex,True,True
|
||||
_conj,private,False,True
|
||||
conj,core complex,True,True
|
||||
conj,core complex,True,False
|
||||
_conj_physical,private,False,False
|
||||
conj_physical,core complex,True,True
|
||||
conj_physical.out,out,False,False
|
||||
conj_physical_,inplace,False,False
|
||||
resolve_conj,core complex,True,False
|
||||
resolve_neg,core unknown,True,False
|
||||
resolve_conj,core complex,True,True
|
||||
resolve_neg,core unknown,True,True
|
||||
_neg_view,private,False,False
|
||||
acos,core primitive pointwise,False,True
|
||||
acos_,inplace,False,True
|
||||
acos.out,out,False,False
|
||||
arccos,core alias,True,False
|
||||
arccos,core alias,True,True
|
||||
arccos_,inplace,True,False
|
||||
arccos.out,out,True,False
|
||||
avg_pool1d,pool,True,False
|
||||
adaptive_avg_pool1d,pool,True,True
|
||||
adaptive_avg_pool1d,pool,True,False
|
||||
adaptive_max_pool1d,pool,True,False
|
||||
add.Tensor,core primitive pointwise,False,True
|
||||
add_.Tensor,inplace,False,True
|
||||
@ -96,7 +96,7 @@ add_.Scalar,inplace,False,True
|
||||
addmv,core composite matmul,False,True
|
||||
addmv_,inplace,False,False
|
||||
addmv.out,out,False,False
|
||||
addr,core composite pointwise,True,True
|
||||
addr,core composite pointwise,True,False
|
||||
addr_,inplace,False,False
|
||||
addr.out,out,True,False
|
||||
affine_grid_generator,core factory,False,False
|
||||
@ -123,19 +123,19 @@ argmin.out,out,False,False
|
||||
acosh,core primitive pointwise,False,True
|
||||
acosh_,inplace,False,True
|
||||
acosh.out,out,False,False
|
||||
arccosh,core alias,True,False
|
||||
arccosh,core alias,True,True
|
||||
arccosh_,inplace,True,False
|
||||
arccosh.out,out,True,False
|
||||
asinh,core primitive pointwise,False,True
|
||||
asinh_,inplace,False,True
|
||||
asinh.out,out,False,False
|
||||
arcsinh,core alias,True,False
|
||||
arcsinh,core alias,True,True
|
||||
arcsinh_,inplace,True,False
|
||||
arcsinh.out,out,True,False
|
||||
atanh,core primitive pointwise,False,True
|
||||
atanh_,inplace,False,True
|
||||
atanh.out,out,False,False
|
||||
arctanh,core alias,True,False
|
||||
arctanh,core alias,True,True
|
||||
arctanh_,inplace,True,False
|
||||
arctanh.out,out,True,False
|
||||
as_strided,core view/reshape,False,True
|
||||
@ -143,24 +143,23 @@ as_strided_,inplace,False,False
|
||||
asin,core primitive pointwise,False,True
|
||||
asin_,inplace,False,True
|
||||
asin.out,out,False,False
|
||||
arcsin,core alias,True,False
|
||||
arcsin,core alias,True,True
|
||||
arcsin_,inplace,True,False
|
||||
arcsin.out,out,True,False
|
||||
atan,core primitive pointwise,False,True
|
||||
atan_,inplace,False,True
|
||||
atan.out,out,False,False
|
||||
arctan,core alias,True,False
|
||||
arctan,core alias,True,True
|
||||
arctan_,inplace,True,False
|
||||
arctan.out,out,True,False
|
||||
atleast_1d,core view/reshape,True,False
|
||||
atleast_1d.Sequence,core view/reshape,True,False
|
||||
atleast_2d,core view/reshape,True,False
|
||||
atleast_2d.Sequence,core view/reshape,True,False
|
||||
atleast_3d,core view/reshape,True,False
|
||||
atleast_3d.Sequence,core view/reshape,True,False
|
||||
atleast_1d,core view/reshape,True,True
|
||||
atleast_1d.Sequence,core view/reshape,True,True
|
||||
atleast_2d,core view/reshape,True,True
|
||||
atleast_2d.Sequence,core view/reshape,True,True
|
||||
atleast_3d,core view/reshape,True,True
|
||||
atleast_3d.Sequence,core view/reshape,True,True
|
||||
baddbmm,core composite matmul,False,True
|
||||
baddbmm_,inplace,False,False
|
||||
_baddbmm_mkl_,inplace,True,False
|
||||
baddbmm.out,out,False,False
|
||||
bartlett_window,core factory,True,False
|
||||
bartlett_window.periodic,core factory,True,False
|
||||
@ -207,15 +206,15 @@ blackman_window.periodic,core factory,True,False
|
||||
bmm,core composite matmul,False,True
|
||||
bmm.out,out,False,False
|
||||
broadcast_tensors,core view/reshape,True,True
|
||||
broadcast_to,core view/reshape,True,False
|
||||
broadcast_to,core view/reshape,True,True
|
||||
cat,core view/reshape,False,True
|
||||
cat.out,out,False,False
|
||||
cat.names,core view/reshape,True,False
|
||||
cat.names_out,out,True,False
|
||||
concat,core unknown,True,True
|
||||
concat.out,out,True,True
|
||||
concat.names,core unknown,True,True
|
||||
concat.names_out,out,True,True
|
||||
concat.out,out,True,False
|
||||
concat.names,core unknown,True,False
|
||||
concat.names_out,out,True,False
|
||||
block_diag,core view/reshape,True,False
|
||||
ceil,core primitive pointwise,False,True
|
||||
ceil_,inplace,False,True
|
||||
@ -235,14 +234,14 @@ clamp.out,out,False,False
|
||||
clamp.Tensor_out,out,False,False
|
||||
clamp_max,core composite pointwise,False,True
|
||||
clamp_max.Tensor,core composite pointwise,False,True
|
||||
clamp_max_,inplace,False,False
|
||||
clamp_max_.Tensor,inplace,False,False
|
||||
clamp_max_,inplace,False,True
|
||||
clamp_max_.Tensor,inplace,False,True
|
||||
clamp_max.out,out,False,False
|
||||
clamp_max.Tensor_out,out,False,False
|
||||
clamp_min,core composite pointwise,False,True
|
||||
clamp_min.Tensor,core composite pointwise,False,True
|
||||
clamp_min_,inplace,False,False
|
||||
clamp_min_.Tensor,inplace,False,False
|
||||
clamp_min_,inplace,False,True
|
||||
clamp_min_.Tensor,inplace,False,True
|
||||
clamp_min.out,out,False,False
|
||||
clamp_min.Tensor_out,out,False,False
|
||||
clip,core alias,True,True
|
||||
@ -366,7 +365,7 @@ div.Scalar,core primitive pointwise,False,True
|
||||
div_.Scalar,inplace,False,True
|
||||
div.Scalar_mode,core primitive pointwise,False,True
|
||||
div_.Scalar_mode,inplace,False,False
|
||||
divide.Tensor,core alias,True,False
|
||||
divide.Tensor,core alias,True,True
|
||||
divide_.Tensor,inplace,True,False
|
||||
divide.out,out,True,False
|
||||
divide.Scalar,core alias,True,False
|
||||
@ -379,7 +378,7 @@ divide_.Scalar_mode,inplace,True,False
|
||||
true_divide.Tensor,core primitive pointwise,True,True
|
||||
true_divide_.Tensor,inplace,True,False
|
||||
true_divide.out,out,True,False
|
||||
true_divide.Scalar,core primitive pointwise,True,True
|
||||
true_divide.Scalar,core primitive pointwise,True,False
|
||||
true_divide_.Scalar,inplace,True,False
|
||||
dot,core reduction,False,True
|
||||
dot.out,out,False,False
|
||||
@ -405,7 +404,7 @@ _embedding_bag_per_sample_weights_backward,backward,False,False
|
||||
empty.names,core factory,True,False
|
||||
empty.memory_format,core factory,False,False
|
||||
new_empty,core factory,False,False
|
||||
new_empty_strided,core factory,True,False
|
||||
new_empty_strided,core factory,False,False
|
||||
new_full,core factory,True,False
|
||||
new_zeros,core factory,True,False
|
||||
new_ones,core factory,True,False
|
||||
@ -414,7 +413,7 @@ _empty_per_channel_affine_quantized,private,False,False
|
||||
resize_,inplace,False,True
|
||||
empty_quantized,core factory,False,False
|
||||
empty.out,out,True,False
|
||||
empty_like,core factory,True,True
|
||||
empty_like,core factory,False,True
|
||||
empty_strided,core factory,False,False
|
||||
erf,core primitive pointwise,False,True
|
||||
erf_,inplace,False,True
|
||||
@ -484,7 +483,7 @@ kaiser_window,core factory,True,False
|
||||
kaiser_window.periodic,core factory,True,False
|
||||
kaiser_window.beta,core factory,True,False
|
||||
hinge_embedding_loss,core reduction,True,False
|
||||
group_norm,core reduction,True,True
|
||||
group_norm,core reduction,True,False
|
||||
native_group_norm,backend,True,True
|
||||
native_group_norm_backward,backward,False,False
|
||||
_fft_r2c,private,False,True
|
||||
@ -503,7 +502,7 @@ index_copy,core scatter/gather,True,False
|
||||
index_copy_.dimname,inplace,True,False
|
||||
index_copy.dimname,core scatter/gather,True,False
|
||||
index_put_,inplace,False,True
|
||||
index_put,core scatter/gather,True,True
|
||||
index_put,core scatter/gather,True,False
|
||||
_index_put_impl_,inplace,False,False
|
||||
instance_norm,core reduction,True,False
|
||||
inverse,core linalg,False,True
|
||||
@ -542,7 +541,7 @@ nan_to_num,core composite pointwise,False,True
|
||||
nan_to_num_,inplace,False,True
|
||||
nan_to_num.out,out,False,False
|
||||
linear,core composite matmul,True,True
|
||||
linear.out,out,True,True
|
||||
linear.out,out,True,False
|
||||
mkldnn_linear,backend,False,False
|
||||
mkldnn_linear_backward_input,backward,False,False
|
||||
mkldnn_linear_backward_weights,backward,False,False
|
||||
@ -610,11 +609,11 @@ matrix_rank.tol,core linalg,True,False
|
||||
matrix_rank,core linalg,True,False
|
||||
matrix_power,core alias,True,False
|
||||
matrix_power.out,out,True,False
|
||||
matrix_exp,core linalg,False,True
|
||||
matrix_exp,core linalg,True,True
|
||||
matrix_exp_backward,backward,True,False
|
||||
_aminmax,private,False,False
|
||||
_aminmax.dim,private,False,False
|
||||
aminmax,core unknown,False,False
|
||||
aminmax,core unknown,False,True
|
||||
aminmax.out,out,False,False
|
||||
_compute_linear_combination,private,False,False
|
||||
_compute_linear_combination.out,out,False,False
|
||||
@ -640,8 +639,8 @@ mean.dim,core reduction,False,True
|
||||
mean.out,out,False,False
|
||||
mean.names_dim,core reduction,True,False
|
||||
mean.names_out,out,True,False
|
||||
nanmean,core unknown,True,True
|
||||
nanmean.out,out,True,True
|
||||
nanmean,core unknown,True,False
|
||||
nanmean.out,out,True,False
|
||||
median,core reduction,False,True
|
||||
median.dim,core reduction,False,True
|
||||
median.dim_values,out,False,False
|
||||
@ -693,7 +692,7 @@ mul_.Tensor,inplace,False,True
|
||||
mul.out,out,False,False
|
||||
mul.Scalar,core reduction,False,True
|
||||
mul_.Scalar,inplace,False,True
|
||||
multiply.Tensor,core alias,True,False
|
||||
multiply.Tensor,core alias,True,True
|
||||
multiply_.Tensor,inplace,True,False
|
||||
multiply.out,out,True,False
|
||||
multiply.Scalar,core alias,True,False
|
||||
@ -729,20 +728,24 @@ ones,core factory,True,False
|
||||
ones.out,out,True,False
|
||||
ones_like,core factory,True,True
|
||||
pairwise_distance,core reduction,True,False
|
||||
cdist,core reduction,True,True
|
||||
cdist,core reduction,True,False
|
||||
_euclidean_dist,private,False,False
|
||||
_cdist_forward,private,False,False
|
||||
_cdist_backward,backward,False,False
|
||||
pdist,core reduction,True,False
|
||||
_pdist_forward,private,False,False
|
||||
_pdist_backward,backward,False,False
|
||||
cosine_similarity,core reduction,True,False
|
||||
cosine_similarity,core reduction,True,True
|
||||
permute,core scatter/gather,False,True
|
||||
movedim.intlist,core view/reshape,True,True
|
||||
movedim.int,core view/reshape,True,True
|
||||
moveaxis.intlist,core alias,True,False
|
||||
moveaxis.intlist,core alias,True,True
|
||||
moveaxis.int,core alias,True,False
|
||||
numpy_T,core view/reshape,True,True
|
||||
matrix_H,core unknown,True,False
|
||||
mT,core unknown,True,True
|
||||
mH,core unknown,True,True
|
||||
adjoint,core unknown,True,False
|
||||
pixel_shuffle,core view/reshape,True,False
|
||||
pixel_unshuffle,core view/reshape,True,False
|
||||
channel_shuffle,core view/reshape,False,False
|
||||
@ -789,14 +792,14 @@ randperm.generator_out,out,False,False
|
||||
range.step,core factory,True,False
|
||||
range,core factory,True,False
|
||||
range.out,out,False,False
|
||||
ravel,core view/reshape,True,False
|
||||
ravel,core view/reshape,True,True
|
||||
reciprocal,core composite pointwise,False,True
|
||||
reciprocal_,inplace,False,True
|
||||
reciprocal.out,out,False,False
|
||||
neg,core composite pointwise,False,True
|
||||
neg_,inplace,False,True
|
||||
neg.out,out,False,False
|
||||
negative,core alias,True,False
|
||||
negative,core alias,True,True
|
||||
negative_,inplace,True,False
|
||||
negative.out,out,True,False
|
||||
repeat,core view/reshape,False,True
|
||||
@ -966,8 +969,8 @@ one_hot,core scatter/gather,True,True
|
||||
flip,core view/reshape,False,True
|
||||
fliplr,core view/reshape,True,True
|
||||
flipud,core view/reshape,True,True
|
||||
roll,core view/reshape,False,False
|
||||
rot90,core view/reshape,False,False
|
||||
roll,core view/reshape,False,True
|
||||
rot90,core view/reshape,False,True
|
||||
trapezoid.x,core reduction,True,True
|
||||
trapezoid.dx,core reduction,True,True
|
||||
trapz.x,core composite pointwise,True,True
|
||||
@ -977,10 +980,10 @@ triplet_margin_loss,core reduction,True,False
|
||||
trunc,core composite pointwise,False,True
|
||||
trunc_,inplace,False,True
|
||||
trunc.out,out,False,False
|
||||
fix,core alias,True,False
|
||||
fix,core alias,True,True
|
||||
fix_,inplace,True,False
|
||||
fix.out,out,True,False
|
||||
type_as,core misc,True,True
|
||||
type_as,core misc,True,False
|
||||
_has_compatible_shallow_copy_type,private,True,False
|
||||
_unique,private,False,False
|
||||
unique_dim,core misc,False,False
|
||||
@ -1007,10 +1010,10 @@ var_mean.names_dim,core reduction,True,False
|
||||
var_mean.correction_names,core reduction,True,False
|
||||
view_as,core view/reshape,True,True
|
||||
where.self,core misc,True,True
|
||||
where.ScalarSelf,core misc,True,True
|
||||
where.ScalarOther,core misc,True,True
|
||||
where.Scalar,core misc,True,True
|
||||
where,core misc,True,True
|
||||
where.ScalarSelf,core misc,True,False
|
||||
where.ScalarOther,core misc,True,False
|
||||
where.Scalar,core misc,True,False
|
||||
where,core misc,True,False
|
||||
_s_where,private,False,True
|
||||
norm_except_dim,core reduction,True,False
|
||||
_weight_norm,private,True,False
|
||||
@ -1072,7 +1075,7 @@ sub_.Tensor,inplace,False,True
|
||||
sub.Scalar,core primitive pointwise,False,True
|
||||
sub_.Scalar,inplace,False,True
|
||||
subtract.out,out,True,False
|
||||
subtract.Tensor,core alias,True,False
|
||||
subtract.Tensor,core alias,True,True
|
||||
subtract_.Tensor,inplace,True,False
|
||||
subtract.Scalar,core alias,True,False
|
||||
subtract_.Scalar,inplace,True,False
|
||||
@ -1165,7 +1168,7 @@ to.dtype_layout,core misc,True,True
|
||||
to.device,core misc,True,True
|
||||
to.dtype,core misc,True,True
|
||||
to.other,core misc,True,True
|
||||
meshgrid,core view/reshape,True,True
|
||||
meshgrid,core view/reshape,True,False
|
||||
meshgrid.indexing,core view/reshape,True,True
|
||||
cartesian_prod,core misc,True,False
|
||||
combinations,core misc,True,False
|
||||
@ -1236,15 +1239,15 @@ scatter.src_out,out,False,False
|
||||
scatter.value,core scatter/gather,False,True
|
||||
scatter_.value,inplace,False,False
|
||||
scatter.value_out,out,False,False
|
||||
scatter.reduce,core scatter/gather,False,False
|
||||
scatter.reduce,core scatter/gather,False,True
|
||||
scatter_.reduce,inplace,False,False
|
||||
scatter.reduce_out,out,False,False
|
||||
scatter.value_reduce,core scatter/gather,False,False
|
||||
scatter.value_reduce,core scatter/gather,False,True
|
||||
scatter_.value_reduce,inplace,False,False
|
||||
scatter.value_reduce_out,out,False,False
|
||||
scatter.dimname_src,core scatter/gather,True,False
|
||||
scatter.dimname_value,core scatter/gather,True,False
|
||||
scatter_add,core scatter/gather,False,False
|
||||
scatter_add,core scatter/gather,False,True
|
||||
scatter_add_,inplace,False,False
|
||||
scatter_add.out,out,False,False
|
||||
scatter_add.dimname,core scatter/gather,True,False
|
||||
@ -1322,7 +1325,7 @@ diag.out,out,False,False
|
||||
diag,core view/reshape,False,True
|
||||
diag_backward,backward,True,False
|
||||
cross.out,out,False,False
|
||||
cross,core misc,False,False
|
||||
cross,core misc,False,True
|
||||
triu.out,out,False,False
|
||||
triu,core view/reshape,False,True
|
||||
tril.out,out,False,False
|
||||
@ -1340,7 +1343,7 @@ ne_.Tensor,inplace,False,False
|
||||
not_equal.Scalar_out,out,True,False
|
||||
not_equal.Scalar,core alias,True,False
|
||||
not_equal.Tensor_out,out,True,False
|
||||
not_equal.Tensor,core alias,True,False
|
||||
not_equal.Tensor,core alias,True,True
|
||||
not_equal_.Scalar,inplace,True,False
|
||||
not_equal_.Tensor,inplace,True,False
|
||||
eq.Scalar_out,out,False,False
|
||||
@ -1356,7 +1359,7 @@ ge_.Tensor,inplace,False,False
|
||||
greater_equal.Scalar_out,out,True,False
|
||||
greater_equal.Scalar,core alias,True,False
|
||||
greater_equal.Tensor_out,out,True,False
|
||||
greater_equal.Tensor,core alias,True,False
|
||||
greater_equal.Tensor,core alias,True,True
|
||||
greater_equal_.Scalar,inplace,True,False
|
||||
greater_equal_.Tensor,inplace,True,False
|
||||
le.Scalar_out,out,False,False
|
||||
@ -1368,7 +1371,7 @@ le_.Tensor,inplace,False,False
|
||||
less_equal.Scalar_out,out,True,False
|
||||
less_equal.Scalar,core alias,True,False
|
||||
less_equal.Tensor_out,out,True,False
|
||||
less_equal.Tensor,core alias,True,False
|
||||
less_equal.Tensor,core alias,True,True
|
||||
less_equal_.Scalar,inplace,True,False
|
||||
less_equal_.Tensor,inplace,True,False
|
||||
gt.Scalar_out,out,False,False
|
||||
@ -1380,7 +1383,7 @@ gt_.Tensor,inplace,False,False
|
||||
greater.Scalar_out,out,True,False
|
||||
greater.Scalar,core alias,True,False
|
||||
greater.Tensor_out,out,True,False
|
||||
greater.Tensor,core alias,True,False
|
||||
greater.Tensor,core alias,True,True
|
||||
greater_.Scalar,inplace,True,False
|
||||
greater_.Tensor,inplace,True,False
|
||||
lt.Scalar_out,out,False,False
|
||||
@ -1392,7 +1395,7 @@ lt_.Tensor,inplace,False,False
|
||||
less.Scalar_out,out,True,False
|
||||
less.Scalar,core alias,True,False
|
||||
less.Tensor_out,out,True,False
|
||||
less.Tensor,core alias,True,False
|
||||
less.Tensor,core alias,True,True
|
||||
less_.Scalar,inplace,True,False
|
||||
less_.Tensor,inplace,True,False
|
||||
take.out,out,False,False
|
||||
@ -1400,7 +1403,7 @@ take,core scatter/gather,False,False
|
||||
take_along_dim.out,out,True,False
|
||||
take_along_dim,core scatter/gather,True,False
|
||||
index_select.out,out,False,False
|
||||
index_select,core scatter/gather,False,False
|
||||
index_select,core scatter/gather,False,True
|
||||
index_select.dimname_out,out,True,False
|
||||
index_select.dimname,core scatter/gather,True,False
|
||||
index_select_backward,backward,True,False
|
||||
@ -1422,7 +1425,7 @@ addcmul_,inplace,False,False
|
||||
addcdiv.out,out,False,False
|
||||
addcdiv,core composite pointwise,False,True
|
||||
addcdiv_,inplace,False,False
|
||||
cross_entropy_loss,core reduction,True,True
|
||||
cross_entropy_loss,core reduction,True,False
|
||||
lstsq.X,out,False,False
|
||||
lstsq,core alias,False,False
|
||||
triangular_solve.X,out,False,False
|
||||
@ -1435,9 +1438,9 @@ eig,core linalg,False,False
|
||||
svd.U,out,True,False
|
||||
svd,core linalg,True,True
|
||||
_svd_helper,private,False,True
|
||||
swapaxes,core alias,True,False
|
||||
swapaxes,core alias,True,True
|
||||
swapaxes_,inplace,True,False
|
||||
swapdims,core alias,True,False
|
||||
swapdims,core alias,True,True
|
||||
swapdims_,inplace,True,False
|
||||
cholesky.out,out,False,False
|
||||
cholesky,core linalg,False,True
|
||||
@ -1497,6 +1500,9 @@ histogram.bins_tensor_out,out,False,False
|
||||
histogram.bins_tensor,core unknown,False,False
|
||||
histogram.bin_ct_out,out,False,False
|
||||
histogram.bin_ct,core unknown,False,False
|
||||
_histogramdd_bin_edges,private,False,False
|
||||
_histogramdd_from_bin_cts,private,False,False
|
||||
_histogramdd_from_bin_tensors,private,False,False
|
||||
fmod.Scalar_out,out,False,False
|
||||
fmod.Scalar,core primitive pointwise,False,True
|
||||
fmod_.Scalar,inplace,False,False
|
||||
@ -1562,7 +1568,7 @@ sort.dimname,core misc,True,False
|
||||
sort.dimname_stable,core misc,True,False
|
||||
msort.out,out,True,False
|
||||
msort,core misc,True,True
|
||||
argsort,core misc,True,False
|
||||
argsort,core misc,True,True
|
||||
argsort.dimname,core misc,True,False
|
||||
topk.values,out,False,False
|
||||
topk,core misc,False,True
|
||||
@ -1587,7 +1593,7 @@ pow_.Tensor,inplace,False,False
|
||||
float_power.Tensor_Tensor_out,out,True,False
|
||||
float_power.Tensor_Tensor,core composite pointwise,True,True
|
||||
float_power.Scalar_out,out,True,False
|
||||
float_power.Scalar,core composite pointwise,True,True
|
||||
float_power.Scalar,core composite pointwise,True,False
|
||||
float_power.Tensor_Scalar_out,out,True,False
|
||||
float_power.Tensor_Scalar,core composite pointwise,True,True
|
||||
float_power_.Scalar,inplace,True,False
|
||||
@ -1702,6 +1708,7 @@ bucketize.Tensor,core misc,False,False
|
||||
bucketize.Tensor_out,out,False,False
|
||||
bucketize.Scalar,core misc,False,False
|
||||
searchsorted.Tensor,core misc,False,False
|
||||
_torch_cuda_cu_linker_symbol_op,private,False,False
|
||||
searchsorted.Tensor_out,out,False,False
|
||||
searchsorted.Scalar,core misc,False,False
|
||||
_convert_indices_from_coo_to_csr,conv,False,False
|
||||
@ -1725,14 +1732,14 @@ multilabel_margin_loss_forward,core reduction,False,False
|
||||
multilabel_margin_loss_backward.grad_input,out,False,False
|
||||
multilabel_margin_loss_backward,backward,False,False
|
||||
nll_loss.out,out,True,False
|
||||
nll_loss_nd,core reduction,True,True
|
||||
nll_loss,core reduction,True,True
|
||||
nll_loss_nd,core reduction,True,False
|
||||
nll_loss,core reduction,True,False
|
||||
nll_loss_forward.output,out,False,False
|
||||
nll_loss_forward,core reduction,False,True
|
||||
nll_loss_backward.grad_input,out,False,False
|
||||
nll_loss_backward,backward,False,True
|
||||
nll_loss2d.out,out,True,False
|
||||
nll_loss2d,core reduction,True,True
|
||||
nll_loss2d,core reduction,True,False
|
||||
nll_loss2d_forward.output,out,False,False
|
||||
nll_loss2d_forward,core reduction,False,False
|
||||
nll_loss2d_backward.grad_input,out,False,False
|
||||
@ -1802,7 +1809,7 @@ mkldnn_adaptive_avg_pool2d_backward,pool,False,False
|
||||
_adaptive_avg_pool2d,pool,False,True
|
||||
_adaptive_avg_pool2d_backward,pool,False,False
|
||||
adaptive_avg_pool3d.out,out,False,False
|
||||
adaptive_avg_pool3d,pool,True,True
|
||||
adaptive_avg_pool3d,pool,True,False
|
||||
_adaptive_avg_pool3d,pool,False,False
|
||||
adaptive_avg_pool3d_backward.grad_input,out,False,False
|
||||
_adaptive_avg_pool3d_backward,pool,False,False
|
||||
@ -1817,7 +1824,7 @@ adaptive_max_pool3d_backward,pool,False,False
|
||||
avg_pool2d.out,out,False,False
|
||||
avg_pool2d,pool,False,True
|
||||
avg_pool2d_backward.grad_input,out,False,False
|
||||
avg_pool2d_backward,pool,False,False
|
||||
avg_pool2d_backward,pool,False,True
|
||||
avg_pool3d.out,out,False,False
|
||||
avg_pool3d,pool,False,False
|
||||
avg_pool3d_backward.grad_input,out,False,False
|
||||
@ -1928,10 +1935,10 @@ slow_conv_transpose3d_backward.grad_output,out,False,False
|
||||
slow_conv_transpose3d_backward.output_mask,conv,False,False
|
||||
thnn_conv2d.out,out,True,False
|
||||
thnn_conv2d,conv,True,False
|
||||
thnn_conv2d_forward.output,out,False,False
|
||||
thnn_conv2d_forward,conv,False,False
|
||||
thnn_conv2d_backward.grad_input,out,False,False
|
||||
thnn_conv2d_backward.output_mask,conv,False,False
|
||||
_slow_conv2d_forward.output,out,False,False
|
||||
_slow_conv2d_forward,conv,False,False
|
||||
_slow_conv2d_backward.grad_input,out,False,False
|
||||
_slow_conv2d_backward.output_mask,conv,False,False
|
||||
_conv_depthwise2d.out,out,False,False
|
||||
_conv_depthwise2d,conv,False,False
|
||||
_conv_depthwise2d_backward.grad_input,out,False,False
|
||||
@ -1960,7 +1967,7 @@ im2col,core view/reshape,False,False
|
||||
im2col_backward.grad_input,out,False,False
|
||||
im2col_backward,backward,False,False
|
||||
isfinite,core composite pointwise,True,True
|
||||
isinf,core composite pointwise,True,True
|
||||
isinf,core composite pointwise,True,False
|
||||
record_stream,out,False,False
|
||||
isposinf,core composite pointwise,False,True
|
||||
isposinf.out,out,False,False
|
||||
@ -2018,11 +2025,11 @@ special_i1,core primitive pointwise,False,True
|
||||
special_i1.out,out,False,False
|
||||
special_i1e,core composite pointwise,False,True
|
||||
special_i1e.out,out,False,False
|
||||
special_logit,core composite pointwise,True,False
|
||||
special_logit,core composite pointwise,True,True
|
||||
special_logit.out,out,True,False
|
||||
special_polygamma,core unknown,True,True
|
||||
special_polygamma.out,out,True,False
|
||||
special_logsumexp,core unknown,True,False
|
||||
special_logsumexp,core unknown,True,True
|
||||
special_logsumexp.out,out,True,False
|
||||
special_expit,core composite pointwise,True,True
|
||||
special_expit.out,out,True,False
|
||||
@ -2032,13 +2039,14 @@ special_round,core primitive pointwise,True,True
|
||||
special_round.out,out,True,False
|
||||
special_log1p,core primitive pointwise,True,True
|
||||
special_log1p.out,out,True,False
|
||||
special_log_softmax,core unknown,True,False
|
||||
special_gammainc.out,out,True,True
|
||||
special_log_softmax,core unknown,True,True
|
||||
special_gammainc.out,out,True,False
|
||||
special_gammainc,core unknown,True,True
|
||||
special_gammaincc.out,out,True,True
|
||||
special_gammaincc.out,out,True,False
|
||||
special_gammaincc,core unknown,True,True
|
||||
special_multigammaln,core unknown,True,False
|
||||
special_multigammaln,core unknown,True,True
|
||||
special_multigammaln.out,out,True,False
|
||||
special_softmax,core unknown,True,True
|
||||
fft_fft,core fft,True,True
|
||||
fft_fft.out,out,True,False
|
||||
fft_ifft,core fft,True,True
|
||||
@ -2059,6 +2067,10 @@ fft_rfft2,core fft,True,False
|
||||
fft_rfft2.out,out,True,False
|
||||
fft_irfft2,core fft,True,False
|
||||
fft_irfft2.out,out,True,False
|
||||
fft_hfft2,core unknown,True,False
|
||||
fft_hfft2.out,out,True,False
|
||||
fft_ihfft2,core unknown,True,False
|
||||
fft_ihfft2.out,out,True,False
|
||||
fft_fftn,core fft,True,False
|
||||
fft_fftn.out,out,True,False
|
||||
fft_ifftn,core fft,True,False
|
||||
@ -2067,6 +2079,10 @@ fft_rfftn,core fft,True,True
|
||||
fft_rfftn.out,out,True,False
|
||||
fft_irfftn,core fft,True,True
|
||||
fft_irfftn.out,out,True,False
|
||||
fft_hfftn,core unknown,True,False
|
||||
fft_hfftn.out,out,True,False
|
||||
fft_ihfftn,core unknown,True,False
|
||||
fft_ihfftn.out,out,True,False
|
||||
fft_fftfreq,core fft,True,False
|
||||
fft_fftfreq.out,out,True,False
|
||||
fft_rfftfreq,core fft,True,False
|
||||
@ -2078,14 +2094,15 @@ linalg_cholesky_ex.L,out,False,False
|
||||
linalg_cholesky,core alias,True,False
|
||||
linalg_cholesky.out,out,True,False
|
||||
linalg_det,core linalg,True,True
|
||||
linalg_det.out,out,True,True
|
||||
linalg_det.out,out,True,False
|
||||
det,core alias,True,True
|
||||
_det_lu_based_helper,private,False,True
|
||||
_det_lu_based_helper_backward_helper,backward,False,False
|
||||
linalg_lstsq,core linalg,False,False
|
||||
linalg_lstsq.out,out,False,False
|
||||
linalg_matmul,core unknown,True,True
|
||||
linalg_matmul.out,out,True,True
|
||||
linalg_matmul.out,out,True,False
|
||||
linalg_matrix_exp,core unknown,False,False
|
||||
linalg_slogdet,core alias,False,True
|
||||
linalg_slogdet.out,out,False,False
|
||||
linalg_eig,core alias,False,True
|
||||
@ -2127,6 +2144,10 @@ linalg_cond,core linalg,True,True
|
||||
linalg_cond.out,out,True,False
|
||||
linalg_cond.p_str,core linalg,True,False
|
||||
linalg_cond.p_str_out,out,True,False
|
||||
linalg_pinv.atol_rtol_tensor,core alias,False,False
|
||||
linalg_pinv.atol_rtol_tensor_out,out,False,False
|
||||
linalg_pinv.atol_rtol_float,core alias,True,False
|
||||
linalg_pinv.atol_rtol_float_out,out,True,False
|
||||
linalg_pinv,core alias,True,True
|
||||
linalg_pinv.rcond_tensor,core alias,True,False
|
||||
linalg_pinv.out,out,True,False
|
||||
@ -2142,6 +2163,10 @@ linalg_qr.out,out,False,False
|
||||
_linalg_qr_helper,private,False,False
|
||||
linalg_matrix_power,core linalg,True,False
|
||||
linalg_matrix_power.out,out,True,False
|
||||
linalg_matrix_rank.atol_rtol_tensor,core alias,True,False
|
||||
linalg_matrix_rank.atol_rtol_tensor_out,out,True,False
|
||||
linalg_matrix_rank.atol_rtol_float,core alias,True,False
|
||||
linalg_matrix_rank.atol_rtol_float_out,out,True,False
|
||||
linalg_matrix_rank,core alias,True,False
|
||||
linalg_matrix_rank.out,out,True,False
|
||||
linalg_matrix_rank.tol_tensor,core alias,True,False
|
||||
@ -2155,6 +2180,7 @@ _test_optional_floatlist,private,False,False
|
||||
_test_string_default,private,True,False
|
||||
_test_ambiguous_defaults.a,private,True,False
|
||||
_test_ambiguous_defaults.b,private,True,False
|
||||
_test_warn_in_autograd,private,False,False
|
||||
segment_reduce,core misc,False,False
|
||||
_segment_reduce_backward,backward,False,False
|
||||
pad_sequence,core misc,True,False
|
||||
|
||||
Reference in New Issue
Block a user