/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
include
/
ATen
/
/usr/local/lib64/python3.6/site-packages/torch/include/ATen
mkdir
upload
Name
Size
Mode
Actions
core/
-
0755
rm
cpu/
-
0755
rm
cuda/
-
0755
rm
cudnn/
-
0755
rm
detail/
-
0755
rm
hip/
-
0755
rm
native/
-
0755
rm
quantized/
-
0755
rm
AccumulateType.h
4438
0644
edit
dl
rm
ArrayRef.h
44
0644
edit
dl
rm
ATen.h
998
0644
edit
dl
rm
autocast_mode.h
6716
0644
edit
dl
rm
Backend.h
43
0644
edit
dl
rm
Backtrace.h
46
0644
edit
dl
rm
BatchedFallback.h
965
0644
edit
dl
rm
BatchedTensorImpl.h
5383
0644
edit
dl
rm
CompositeExplicitAutogradFunctions.h
1622
0644
edit
dl
rm
CompositeExplicitAutogradFunctions_inl.h
54075
0644
edit
dl
rm
CompositeImplicitAutogradFunctions.h
1622
0644
edit
dl
rm
CompositeImplicitAutogradFunctions_inl.h
142082
0644
edit
dl
rm
Config.h
734
0644
edit
dl
rm
Context.h
12767
0644
edit
dl
rm
cpp_custom_type_hack.h
5326
0644
edit
dl
rm
CPUApplyUtils.h
12582
0644
edit
dl
rm
CPUFixedAllocator.h
830
0644
edit
dl
rm
CPUFunctions.h
1600
0644
edit
dl
rm
CPUFunctions_inl.h
171924
0644
edit
dl
rm
CPUGeneratorImpl.h
1431
0644
edit
dl
rm
CUDAFunctions.h
1601
0644
edit
dl
rm
CUDAFunctions_inl.h
185696
0644
edit
dl
rm
CUDAGeneratorImpl.h
4695
0644
edit
dl
rm
Device.h
42
0644
edit
dl
rm
DeviceGuard.h
1134
0644
edit
dl
rm
Dimname.h
31
0644
edit
dl
rm
DimVector.h
46
0644
edit
dl
rm
Dispatch.h
52137
0644
edit
dl
rm
div_rtn.h
204
0644
edit
dl
rm
DLConvertor.h
576
0644
edit
dl
rm
dlpack.h
5244
0644
edit
dl
rm
DynamicLibrary.h
369
0644
edit
dl
rm
ExpandUtils.h
14506
0644
edit
dl
rm
Formatting.h
34
0644
edit
dl
rm
Functions.h
846326
0644
edit
dl
rm
Generator.h
46
0644
edit
dl
rm
InferSize.h
2143
0644
edit
dl
rm
InitialTensorOptions.h
445
0644
edit
dl
rm
Layout.h
42
0644
edit
dl
rm
MapAllocator.h
2999
0644
edit
dl
rm
MatrixRef.h
3016
0644
edit
dl
rm
MemoryOverlap.h
1117
0644
edit
dl
rm
MetaFunctions.h
1601
0644
edit
dl
rm
MetaFunctions_inl.h
84006
0644
edit
dl
rm
NamedTensor.h
35
0644
edit
dl
rm
NamedTensorUtils.h
5747
0644
edit
dl
rm
NativeFunctions.h
354651
0644
edit
dl
rm
NativeMetaFunctions.h
35445
0644
edit
dl
rm
NumericUtils.h
2787
0644
edit
dl
rm
OpaqueTensorImpl.h
6080
0644
edit
dl
rm
Operators.h
1707199
0644
edit
dl
rm
OpMathType.h
460
0644
edit
dl
rm
Parallel.h
4875
0644
edit
dl
rm
ParallelNative.h
2443
0644
edit
dl
rm
ParallelNativeTBB.h
2934
0644
edit
dl
rm
ParallelOpenMP.h
3049
0644
edit
dl
rm
PTThreadPool.h
394
0644
edit
dl
rm
record_function.h
24044
0644
edit
dl
rm
RedispatchFunctions.h
1112886
0644
edit
dl
rm
RegistrationDeclarations.h
545777
0644
edit
dl
rm
SavedTensorHooks.h
328
0644
edit
dl
rm
Scalar.h
44
0644
edit
dl
rm
ScalarOps.h
2272
0644
edit
dl
rm
ScalarType.h
129
0644
edit
dl
rm
SequenceNumber.h
373
0644
edit
dl
rm
SmallVector.h
47
0644
edit
dl
rm
SparseCsrTensorImpl.h
2045
0644
edit
dl
rm
SparseCsrTensorUtils.h
523
0644
edit
dl
rm
SparseTensorImpl.h
12417
0644
edit
dl
rm
SparseTensorUtils.h
4219
0644
edit
dl
rm
Storage.h
43
0644
edit
dl
rm
Tensor.h
48
0644
edit
dl
rm
TensorAccessor.h
51
0644
edit
dl
rm
TensorGeometry.h
1855
0644
edit
dl
rm
TensorIndexing.h
21923
0644
edit
dl
rm
TensorIterator.h
29962
0644
edit
dl
rm
TensorIteratorInternal.h
1862
0644
edit
dl
rm
TensorMeta.h
2917
0644
edit
dl
rm
TensorNames.h
2519
0644
edit
dl
rm
TensorOperators.h
3275
0644
edit
dl
rm
TensorOptions.h
49
0644
edit
dl
rm
TensorUtils.h
5687
0644
edit
dl
rm
ThreadLocalState.h
3289
0644
edit
dl
rm
TracerMode.h
5576
0644
edit
dl
rm
TypeDefault.h
680
0644
edit
dl
rm
Utils.h
5993
0644
edit
dl
rm
Version.h
340
0644
edit
dl
rm
VmapMode.h
952
0644
edit
dl
rm
VmapTransforms.h
7654
0644
edit
dl
rm
WrapDimUtils.h
3438
0644
edit
dl
rm
WrapDimUtilsMulti.h
768
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/include/ATen/NativeMetaFunctions.h
(35445B)
#pragma once // @generated by tools/codegen/gen.py from NativeMetaFunctions.h #include <ATen/TensorMeta.h> #include <ATen/TensorIterator.h> namespace at { namespace meta { struct TORCH_API structured_sgn : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_acos : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_add_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other, const at::Scalar & alpha); }; struct TORCH_API structured_addmv : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Tensor & mat, const at::Tensor & vec, const at::Scalar & beta, const at::Scalar & alpha); }; struct TORCH_API structured_all_dim : public at::impl::MetaBase { template <bool DIM = false> struct TORCH_API precompute_out { precompute_out<true> set_dim(int64_t value) { static_assert(DIM == false, "dim already set"); precompute_out<true> ret; ret.dim = value; return ret; } int64_t dim; }; using meta_return_ty = precompute_out <true>; meta_return_ty meta(const at::Tensor & self, int64_t dim, bool keepdim); }; struct TORCH_API structured_any_dim : public at::impl::MetaBase { template <bool DIM = false> struct TORCH_API precompute_out { precompute_out<true> set_dim(int64_t value) { static_assert(DIM == false, "dim already set"); precompute_out<true> ret; ret.dim = value; return ret; } int64_t dim; }; using meta_return_ty = precompute_out <true>; meta_return_ty meta(const at::Tensor & self, int64_t dim, bool keepdim); }; struct TORCH_API structured_argmax : public at::impl::MetaBase { void meta(const at::Tensor & self, c10::optional<int64_t> dim, bool keepdim); }; struct TORCH_API structured_argmin : public at::impl::MetaBase { void meta(const at::Tensor & self, c10::optional<int64_t> dim, bool keepdim); }; struct TORCH_API structured_acosh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_asinh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_atanh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_asin : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_atan : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_bitwise_not : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_copysign_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_ceil : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_clamp : public TensorIteratorBase { void meta(const at::Tensor & self, at::OptionalScalarRef min, at::OptionalScalarRef max); }; struct TORCH_API structured_cos : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_cosh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_cumprod : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, c10::optional<at::ScalarType> dtype); }; struct TORCH_API structured_cumsum : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, c10::optional<at::ScalarType> dtype); }; struct TORCH_API structured_div_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_div_Tensor_mode : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other, c10::optional<c10::string_view> rounding_mode); }; struct TORCH_API structured_erf : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_erfc : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_exp : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_exp2 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_expm1 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_floor : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_frac : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_gcd : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_lcm : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_isin_Tensor_Tensor : public at::impl::MetaBase { void meta(const at::Tensor & elements, const at::Tensor & test_elements, bool assume_unique, bool invert); }; struct TORCH_API structured_isin_Tensor_Scalar : public at::impl::MetaBase { void meta(const at::Tensor & elements, const at::Scalar & test_element, bool assume_unique, bool invert); }; struct TORCH_API structured_isin_Scalar_Tensor : public at::impl::MetaBase { void meta(const at::Scalar & element, const at::Tensor & test_elements, bool assume_unique, bool invert); }; struct TORCH_API structured_log : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_log10 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_log1p : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_log2 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_logaddexp : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_logaddexp2 : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_xlogy_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured__log_softmax : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, bool half_to_float); }; struct TORCH_API structured__log_softmax_backward_data : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & output, int64_t dim, const at::Tensor & self); }; struct TORCH_API structured_aminmax : public at::impl::MetaBase { void meta(const at::Tensor & self, c10::optional<int64_t> dim, bool keepdim); }; struct TORCH_API structured_mean_dim : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef dim, bool keepdim, c10::optional<at::ScalarType> dtype); }; struct TORCH_API structured_mm : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Tensor & mat2); }; struct TORCH_API structured_mul_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_reciprocal : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_neg : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_round : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_gelu : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_gelu_backward : public TensorIteratorBase { void meta(const at::Tensor & grad, const at::Tensor & self); }; struct TORCH_API structured_hardshrink : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & lambd); }; struct TORCH_API structured_hardshrink_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_out, const at::Tensor & self, const at::Scalar & lambd); }; struct TORCH_API structured_rsqrt : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_silu : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_silu_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self); }; struct TORCH_API structured_mish : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_sigmoid : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_sin : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_sinc : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_sinh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured__softmax : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, bool half_to_float); }; struct TORCH_API structured__softmax_backward_data : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & output, int64_t dim, const at::Tensor & self); }; struct TORCH_API structured_sum_dim_IntList : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef dim, bool keepdim, c10::optional<at::ScalarType> dtype); }; struct TORCH_API structured_sqrt : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_prod_dim_int : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, bool keepdim, c10::optional<at::ScalarType> dtype); }; struct TORCH_API structured_tan : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_tanh : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_threshold : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & threshold, const at::Scalar & value); }; struct TORCH_API structured_threshold_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Scalar & threshold); }; struct TORCH_API structured_trunc : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_norm_ScalarOpt_dim_dtype : public at::impl::MetaBase { void meta(const at::Tensor & self, at::OptionalScalarRef p, at::IntArrayRef dim, bool keepdim, at::ScalarType dtype); }; struct TORCH_API structured_norm_ScalarOpt_dim : public at::impl::MetaBase { void meta(const at::Tensor & self, at::OptionalScalarRef p, at::IntArrayRef dim, bool keepdim); }; struct TORCH_API structured_sub_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other, const at::Scalar & alpha); }; struct TORCH_API structured_heaviside : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & values); }; struct TORCH_API structured_addmm : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Tensor & mat1, const at::Tensor & mat2, const at::Scalar & beta, const at::Scalar & alpha); }; struct TORCH_API structured_scatter_src : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src); }; struct TORCH_API structured_scatter_value : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value); }; struct TORCH_API structured_scatter_reduce : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, c10::string_view reduce); }; struct TORCH_API structured_scatter_value_reduce : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, c10::string_view reduce); }; struct TORCH_API structured_scatter_add : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src); }; struct TORCH_API structured_eq_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_eq_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_bitwise_and_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_bitwise_or_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_bitwise_xor_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_bitwise_left_shift_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_bitwise_right_shift_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_digamma : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_ne_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_ne_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_ge_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_ge_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_le_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_le_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_gt_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_gt_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_lt_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & other); }; struct TORCH_API structured_lt_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_gather : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t dim, const at::Tensor & index, bool sparse_grad); }; struct TORCH_API structured_addcmul : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & tensor1, const at::Tensor & tensor2, const at::Scalar & value); }; struct TORCH_API structured_addcdiv : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & tensor1, const at::Tensor & tensor2, const at::Scalar & value); }; struct TORCH_API structured_lgamma : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_polygamma : public TensorIteratorBase { void meta(int64_t n, const at::Tensor & self); }; struct TORCH_API structured_erfinv : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_i0 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_sign : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_signbit : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_atan2 : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_fmod_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_hypot : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_igamma : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_igammac : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_nextafter : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_remainder_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_fmin : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_fmax : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_maximum : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_minimum : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_topk : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t k, int64_t dim, bool largest, bool sorted); }; struct TORCH_API structured_all : public at::impl::MetaBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_any : public at::impl::MetaBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_renorm : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Scalar & p, int64_t dim, const at::Scalar & maxnorm); }; struct TORCH_API structured_pow_Tensor_Tensor : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & exponent); }; struct TORCH_API structured_pow_Scalar : public at::impl::MetaBase { void meta(const at::Scalar & self, const at::Tensor & exponent); }; struct TORCH_API structured_pow_Tensor_Scalar : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & exponent); }; struct TORCH_API structured__convert_indices_from_coo_to_csr : public at::impl::MetaBase { void meta(const at::Tensor & self, int64_t size, bool out_int32); }; struct TORCH_API structured_nll_loss_forward : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Tensor & target, at::OptionalTensorRef weight, int64_t reduction, int64_t ignore_index); }; struct TORCH_API structured_nll_loss_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Tensor & target, at::OptionalTensorRef weight, int64_t reduction, int64_t ignore_index, const at::Tensor & total_weight); }; struct TORCH_API structured_elu : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & alpha, const at::Scalar & scale, const at::Scalar & input_scale); }; struct TORCH_API structured_elu_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Scalar & alpha, const at::Scalar & scale, const at::Scalar & input_scale, bool is_result, const at::Tensor & self_or_result); }; struct TORCH_API structured_glu : public TensorIteratorBase { void meta(const at::Tensor & self, int64_t dim); }; struct TORCH_API structured_hardsigmoid : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_hardsigmoid_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self); }; struct TORCH_API structured_leaky_relu : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & negative_slope); }; struct TORCH_API structured_leaky_relu_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Scalar & negative_slope, bool self_is_result); }; struct TORCH_API structured_softplus : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & beta, const at::Scalar & threshold); }; struct TORCH_API structured_softplus_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Scalar & beta, const at::Scalar & threshold, const at::Tensor & output); }; struct TORCH_API structured_softshrink : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Scalar & lambd); }; struct TORCH_API structured_softshrink_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Scalar & lambd); }; struct TORCH_API structured_adaptive_max_pool2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size); }; struct TORCH_API structured_adaptive_max_pool2d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Tensor & indices); }; struct TORCH_API structured_adaptive_max_pool3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size); }; struct TORCH_API structured_adaptive_max_pool3d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, const at::Tensor & indices); }; struct TORCH_API structured_avg_pool2d : public at::impl::MetaBase { template <bool KH = false, bool KW = false, bool DH = false, bool DW = false, bool PADH = false, bool PADW = false> struct TORCH_API precompute_out { precompute_out<true, KW, DH, DW, PADH, PADW> set_kH(int64_t value) { static_assert(KH == false, "kH already set"); precompute_out<true, KW, DH, DW, PADH, PADW> ret; ret.kH = value; ret.kW = this->kW; ret.dH = this->dH; ret.dW = this->dW; ret.padH = this->padH; ret.padW = this->padW; return ret; } precompute_out<KH, true, DH, DW, PADH, PADW> set_kW(int64_t value) { static_assert(KW == false, "kW already set"); precompute_out<KH, true, DH, DW, PADH, PADW> ret; ret.kH = this->kH; ret.kW = value; ret.dH = this->dH; ret.dW = this->dW; ret.padH = this->padH; ret.padW = this->padW; return ret; } precompute_out<KH, KW, true, DW, PADH, PADW> set_dH(int64_t value) { static_assert(DH == false, "dH already set"); precompute_out<KH, KW, true, DW, PADH, PADW> ret; ret.kH = this->kH; ret.kW = this->kW; ret.dH = value; ret.dW = this->dW; ret.padH = this->padH; ret.padW = this->padW; return ret; } precompute_out<KH, KW, DH, true, PADH, PADW> set_dW(int64_t value) { static_assert(DW == false, "dW already set"); precompute_out<KH, KW, DH, true, PADH, PADW> ret; ret.kH = this->kH; ret.kW = this->kW; ret.dH = this->dH; ret.dW = value; ret.padH = this->padH; ret.padW = this->padW; return ret; } precompute_out<KH, KW, DH, DW, true, PADW> set_padH(int64_t value) { static_assert(PADH == false, "padH already set"); precompute_out<KH, KW, DH, DW, true, PADW> ret; ret.kH = this->kH; ret.kW = this->kW; ret.dH = this->dH; ret.dW = this->dW; ret.padH = value; ret.padW = this->padW; return ret; } precompute_out<KH, KW, DH, DW, PADH, true> set_padW(int64_t value) { static_assert(PADW == false, "padW already set"); precompute_out<KH, KW, DH, DW, PADH, true> ret; ret.kH = this->kH; ret.kW = this->kW; ret.dH = this->dH; ret.dW = this->dW; ret.padH = this->padH; ret.padW = value; return ret; } int64_t kH; int64_t kW; int64_t dH; int64_t dW; int64_t padH; int64_t padW; }; using meta_return_ty = precompute_out <true, true, true, true, true, true>; meta_return_ty meta(const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, bool ceil_mode, bool count_include_pad, c10::optional<int64_t> divisor_override); }; struct TORCH_API structured_avg_pool2d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, bool ceil_mode, bool count_include_pad, c10::optional<int64_t> divisor_override); }; struct TORCH_API structured_avg_pool3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, bool ceil_mode, bool count_include_pad, c10::optional<int64_t> divisor_override); }; struct TORCH_API structured_avg_pool3d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, bool ceil_mode, bool count_include_pad, c10::optional<int64_t> divisor_override); }; struct TORCH_API structured_fractional_max_pool2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef output_size, const at::Tensor & random_samples); }; struct TORCH_API structured_max_pool2d_with_indices : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool ceil_mode); }; struct TORCH_API structured_max_pool2d_with_indices_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool ceil_mode, const at::Tensor & indices); }; struct TORCH_API structured_reflection_pad1d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_reflection_pad1d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_reflection_pad3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_reflection_pad3d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_replication_pad1d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_replication_pad1d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_replication_pad2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_replication_pad3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef padding); }; struct TORCH_API structured_upsample_linear1d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, bool align_corners, c10::optional<double> scales); }; struct TORCH_API structured_upsample_linear1d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional<double> scales); }; struct TORCH_API structured_upsample_bilinear2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, bool align_corners, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_bilinear2d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_bicubic2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, bool align_corners, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_bicubic2d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_trilinear3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, bool align_corners, c10::optional<double> scales_d, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_trilinear3d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional<double> scales_d, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_nearest1d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, c10::optional<double> scales); }; struct TORCH_API structured_upsample_nearest1d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional<double> scales); }; struct TORCH_API structured_upsample_nearest2d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_nearest2d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_nearest3d : public at::impl::MetaBase { void meta(const at::Tensor & self, at::IntArrayRef output_size, c10::optional<double> scales_d, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_upsample_nearest3d_backward : public at::impl::MetaBase { void meta(const at::Tensor & grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional<double> scales_d, c10::optional<double> scales_h, c10::optional<double> scales_w); }; struct TORCH_API structured_sigmoid_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & output); }; struct TORCH_API structured_logit_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & self, c10::optional<double> eps); }; struct TORCH_API structured_tanh_backward : public TensorIteratorBase { void meta(const at::Tensor & grad_output, const at::Tensor & output); }; struct TORCH_API structured_slow_conv_transpose2d : public at::impl::MetaBase { void meta(const at::Tensor & self, const at::Tensor & weight, at::IntArrayRef kernel_size, at::OptionalTensorRef bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef dilation); }; struct TORCH_API structured_isposinf : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_isneginf : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_entr : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_ndtri : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_erfcx : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_xlog1py : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_special_zeta : public TensorIteratorBase { void meta(const at::Tensor & self, const at::Tensor & other); }; struct TORCH_API structured_special_i0e : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_i1 : public TensorIteratorBase { void meta(const at::Tensor & self); }; struct TORCH_API structured_special_i1e : public TensorIteratorBase { void meta(const at::Tensor & self); }; } // namespace meta } // namespace at
Save
cmd:
run