- op : acos args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : acos backward : acos_grad - op : acosh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : acosh backward : acosh_grad - op : angle args : (Tensor x) output : Tensor infer_meta : func : RealAndImagInferMeta kernel : func : angle backward : angle_grad - op : argsort args : (Tensor x, int axis=-1, bool descending=false) output : Tensor(out), Tensor(indices) infer_meta : func : ArgsortInferMeta kernel : func : argsort backward : argsort_grad - op : asin args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : asin backward : asin_grad - op : asinh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : asinh backward : asinh_grad - op : atan args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : atan backward : atan_grad - op : atan2 args : (Tensor x, Tensor y) output : Tensor infer_meta : func : Atan2InferMeta kernel : func : atan2 backward : atan2_grad - op : atanh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : atanh backward : atanh_grad - op : bernoulli args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : bernoulli - op : bmm args : (Tensor x, Tensor y) output : Tensor infer_meta : func : BmmInferMeta kernel : func : bmm backward : bmm_grad - op : ceil args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : ceil inplace : (x -> out) backward : ceil_grad - op : cholesky args : (Tensor x, bool upper=false) output : Tensor infer_meta : func : CholeskyInferMeta kernel : func : cholesky backward : cholesky_grad - op : cholesky_solve args : (Tensor x, Tensor y, bool upper=false) output : Tensor infer_meta : func : CholeskySolveInferMeta kernel : func : cholesky_solve backward : cholesky_solve_grad - op : cos args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : cos backward : cos_grad - op : cosh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : cosh backward : cosh_grad - op : cross args : (Tensor x, Tensor y, int axis = 9) output : Tensor infer_meta : func : CrossInferMeta kernel : func : cross data_type : x backward : cross_grad - op : det args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : determinant backward : det_grad - op : diag args : (Tensor x, int offset = 0, float padding_value = 0.0) output : Tensor infer_meta : func : DiagInferMeta kernel : func : diag backward : diag_grad - op : diagonal args : (Tensor x, int offset = 0, int axis1 = 0, int axis2 = 1) output : Tensor infer_meta : func : DiagonalInferMeta kernel : func : diagonal backward : diagonal_grad - op : digamma args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : digamma backward : digamma_grad - op : dist args : (Tensor x, Tensor y, float p = 2.0) output : Tensor infer_meta : func : DistInferMeta kernel : func : dist backward : dist_grad - op : dot args : (Tensor x, Tensor y) output : Tensor infer_meta : func : DotInferMeta kernel : func : dot data_type : x backward : dot_grad - op : erf args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : erf backward : erf_grad - op : erfinv args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : erfinv inplace : (x -> out) backward : erfinv_grad - op : exp args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : exp inplace : (x -> out) backward : exp_grad - op : expm1 args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta param : [x] kernel : func : expm1 backward : expm1_grad - op : fft_c2c args : (Tensor x, int64_t[] axes, str normalization, bool forward) output : Tensor infer_meta : func : FFTC2CInferMeta kernel : func : fft_c2c backward : fft_c2c_grad - op : fft_c2r args : (Tensor x, int64_t[] axes, str normalization, bool forward, int64_t last_dim_size=0L) output : Tensor infer_meta : func : FFTC2RInferMeta kernel : func : fft_c2r backward : fft_c2r_grad - op : fft_r2c args : (Tensor x, int64_t[] axes, str normalization, bool forward, bool onesided) output : Tensor infer_meta : func : FFTR2CInferMeta kernel : func : fft_r2c backward : fft_r2c_grad - op : flip args : (Tensor x, int[] axis) output : Tensor (out) infer_meta : func : FlipInferMeta kernel : func : flip backward : flip_grad - op : floor args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : floor inplace : (x -> out) backward : floor_grad - op : hardshrink args : (Tensor x, float threshold = 0.5) output : Tensor (out) infer_meta : func : UnchangedInferMeta param : [x] kernel : func : hard_shrink backward : hardshrink_grad - op : hardsigmoid args : (Tensor x, float slope = 0.2, float offset = 0.5) output : Tensor (out) infer_meta : func : UnchangedInferMeta param : [x] kernel : func : hard_sigmoid backward : hardsigmoid_grad - op : lgamma args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : lgamma backward : lgamma_grad - op : log10 args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : log10 backward: log10_grad - op : log1p args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : log1p backward: log1p_grad - op : log2 args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : log2 backward: log2_grad - op : logit args : (Tensor x, float eps = 1e-6f) output : Tensor infer_meta : func : UnchangedInferMeta param : [x] kernel : func : logit backward : logit_grad - op : logsigmoid args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : logsigmoid backward : logsigmoid_grad - op : mv args : (Tensor x, Tensor vec) output : Tensor infer_meta : func : MvInferMeta kernel : func : mv backward : mv_grad - op : poisson args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : poisson backward : poisson_grad - op : reciprocal args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : reciprocal inplace : (x -> out) backward : reciprocal_grad - op : round args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : round inplace : (x -> out) backward : round_grad - op : send_uv args : (Tensor x, Tensor y, Tensor src_index, Tensor dst_index, str message_op = "ADD") output : Tensor(out) infer_meta : func : SendUVInferMeta kernel : func : send_uv data_type : x backward : send_uv_grad - op : silu args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : silu backward : silu_grad - op : sin args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sin backward : sin_grad - op : sinh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sinh backward : sinh_grad - op : solve args : (Tensor x, Tensor y) output : Tensor infer_meta : func : SolveInferMeta kernel : func : solve data_type : x backward : solve_grad - op : tan args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : tan backward : tan_grad - op : tanh args : (Tensor x) output : Tensor(out) infer_meta : func : UnchangedInferMeta kernel : func : tanh inplace : (x -> out) backward : tanh_grad - op : trace args : (Tensor x, int offset = 0, int axis1 = 0, int axis2 = 1) output : Tensor infer_meta : func : TraceInferMeta kernel : func : trace backward : trace_grad - op : trunc args : (Tensor input) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : trunc backward : trunc_grad