#ifndef TH_GENERIC_FILE #define TH_GENERIC_FILE "TH/generic/THVector.h" #else #include #include TH_API void THVector_(fill)(scalar_t *x, const scalar_t c, const ptrdiff_t n); #if !defined(TH_REAL_IS_BOOL) /* non bool only part */ TH_API void THVector_(cadd)(scalar_t *z, const scalar_t *x, const scalar_t *y, const scalar_t c, const ptrdiff_t n); TH_API void THVector_(adds)(scalar_t *y, const scalar_t *x, const scalar_t c, const ptrdiff_t n); TH_API void THVector_(cmul)(scalar_t *z, const scalar_t *x, const scalar_t *y, const ptrdiff_t n); TH_API void THVector_(muls)(scalar_t *y, const scalar_t *x, const scalar_t c, const ptrdiff_t n); TH_API void THVector_(cdiv)(scalar_t *z, const scalar_t *x, const scalar_t *y, const ptrdiff_t n); TH_API void THVector_(divs)(scalar_t *y, const scalar_t *x, const scalar_t c, const ptrdiff_t n); TH_API void THVector_(neg)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(normal_fill)(scalar_t *data, const int64_t size, struct at::Generator *generator, const scalar_t mean, const scalar_t stddev); #endif /* non bool only part */ #if defined(TH_REAL_IS_SHORT) || defined(TH_REAL_IS_INT) || defined(TH_REAL_IS_LONG) TH_API void THVector_(abs)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); #endif /* floating point only now */ #if defined(TH_REAL_IS_FLOAT) || defined(TH_REAL_IS_DOUBLE) TH_API void THVector_(log10)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(log1p)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(log2)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(sigmoid)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(exp)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(erf)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(erfc)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(cos)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(acos)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(cosh)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(sin)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(asin)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(sinh)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(tan)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(atan)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(tanh)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(pow)(scalar_t *y, const scalar_t *x, const scalar_t c, const ptrdiff_t n); TH_API void THVector_(sqrt)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(rsqrt)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(ceil)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(floor)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(round)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(abs)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(trunc)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(frac)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); TH_API void THVector_(cinv)(scalar_t *y, const scalar_t *x, const ptrdiff_t n); #endif /* floating point only part */ #endif