- api : add args : (Tensor x, Tensor y) output : Tensor infer_meta : func : ElementwiseInferMeta kernel : func : add - api : cast args : (Tensor x, DataType out_dtype) output : Tensor infer_meta : func : CastInferMeta kernel : func : cast param : [x, out_dtype] data_type : x - api : concat args : (Tensor[] x, Scalar axis) output : Tensor infer_meta : func : ConcatInferMeta param : [x, axis] kernel : func : concat - api : conj args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : conj - api : copy_to args : (Tensor x, Backend backend, bool blocking) output : Tensor invoke : copy_to_impl(x, backend, blocking) - api : divide args : (Tensor x, Tensor y) output : Tensor infer_meta : func : ElementwiseInferMeta kernel : func : divide - api : dot args : (Tensor x, Tensor y) output : Tensor infer_meta : func : DotInferMeta kernel : func : dot - api : empty args : (ScalarArray shape, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU) output: Tensor infer_meta : func : CreateInferMeta param : [shape, dtype] kernel : func : empty param : [shape, dtype] data_type : dtype backend : place - api : empty_like args : (Tensor x, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED) output: Tensor infer_meta : func : CreateLikeInferMeta param : [x, dtype] kernel : func : empty_like param : [x, dtype] data_type : dtype > x backend : place > x - api : flatten args : (Tensor x, int start_axis, int stop_axis) output : Tensor infer_meta : func : FlattenInferMeta kernel : func : flatten - api : full args : (ScalarArray shape, Scalar value, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU) output: Tensor infer_meta : func : CreateInferMeta param : [shape, dtype] kernel : func : full param : [shape, value, dtype] data_type : dtype backend : place - api : full_like args : (Tensor x, Scalar value, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED) output: Tensor infer_meta : func : CreateLikeInferMeta param : [x, dtype] kernel : func : full_like param : [x, value, dtype] data_type : dtype > x backend : place > x - api : matmul args : (Tensor x, Tensor y, bool transpose_x = false, bool transpose_y = false) output : Tensor infer_meta : func : MatmulInferMeta kernel : func : matmul backward : matmul_grad - api : mean args : (Tensor x, int64[] axis={}, bool keep_dim=false) output : Tensor infer_meta : func : ReduceInferMeta kernel : func : mean - api : multiply args : (Tensor x, Tensor y) output : Tensor infer_meta : func : ElementwiseInferMeta kernel : func : multiply - api : ones_like args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED) output : Tensor invoke : full_like(x, 1, dtype, place) - api : reshape args : (Tensor x, ScalarArray shape) output : Tensor(out) infer_meta : func : ReshapeInferMeta kernel : func : reshape inplace : (x -> out) - api : scale args : (Tensor x, Scalar scale, float bias, bool bias_after_scale) output : Tensor infer_meta : func : UnchangedInferMeta param : [x] kernel : func : scale, scale_sr - api : sign args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sign - api : split args : (Tensor x, ScalarArray num_or_sections, Scalar axis) output : Tensor[] invoke : split_impl(x, num_or_sections, axis) - api : subtract args : (Tensor x, Tensor y) output : Tensor infer_meta : func : ElementwiseInferMeta kernel : func : subtract - api : sum args : (Tensor x, int64[] axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false) output : Tensor infer_meta : func : SumInferMeta kernel : func : sum data_type : x - api : zeros_like args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED) output : Tensor invoke : full_like(x, 0, dtype, place) - api : digamma args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : digamma backward : digamma_grad - api : abs args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : abs backward : abs_grad - api : trunc args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : trunc backward : trunc_grad # - api : norm # args : (Tensor x, int axis, float epsilon, bool is_test) # output : Tensor(out), Tensor(norm) # infer_meta : # func : NormInferMeta # kernel : # func : norm # intermediate : norm # backward : norm_grad - api : diagonal args : (Tensor x, int offset, int axis1, int axis2) output : Tensor infer_meta : func : DiagonalInferMeta kernel : func : diagonal backward : diagonal_grad # softmax - api : softmax args : (Tensor x, int axis) output : Tensor infer_meta : func : SoftmaxInferMeta kernel : func : softmax backward : softmax_grad # maxout - api : maxout args : (Tensor x, int groups, int axis) output : Tensor infer_meta : func : MaxoutInferMeta kernel : func : maxout backward : maxout_grad # put_along_axis - api : put_along_axis args : (Tensor x, Tensor index, Tensor value, int axis, string reduce) output : Tensor infer_meta : func : PutAlongAxisInferMeta kernel : func : put_along_axis backward : put_along_axis_grad # take_along_axis - api : take_along_axis args : (Tensor x, Tensor index, int axis) output : Tensor infer_meta : func : TakeAlongAxisInferMeta kernel : func : take_along_axis backward : take_along_axis_grad # matrix_power - api : maxtrix_power args : (Tensor x, int n) output : Tensor infer_meta : func : MaxtrixPowerInferMeta kernel : func : maxtrix_power backward : maxtrix_power_grad # eigh - api : eigh args : (Tensor x, string uplo) output : Tensor(out_w), Tensor(out_v) infer_meta : func : EighInferMeta kernel : func : eigh backward : eigh_grad # segment_pool - api : segment_pool args : (Tensor x, Tensor segment_ids, string pooltype) output : Tensor(out), Tensor(summed_ids) infer_meta : func : SegmentPoolInferMeta kernel : func : segment_pool backward : segment_pool_grad # accuracy - api : accuracy args : (Tensor x, Tensor indices, Tensor label) output : Tensor(accuracy), Tensor(correct), Tensor(total) infer_meta : func : AccuracyInferMeta kernel : func : accuracy # sin - api : sin args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sin backward : sin_grad # cos - api : cos args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : cos backward : cos_grad # tanh - api : tanh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : tanh backward : tanh_grad # acos - api : acos args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : acos backward : acos_grad # asin - api : asin args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : asin backward : asin_grad # atan - api : atan args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : atan backward : atan_grad # sinh - api : sinh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sinh backward : sinh_grad # cosh - api : cosh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : cosh backward : cosh_grad # asinh - api : asinh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : asinh backward : asinh_grad # acosh - api : acosh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : acosh backward : acosh_grad # atanh - api : atanh args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : atanh backward : atanh_grad # relu - api : relu args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : relu backward : relu_grad # arg_min # int64 ???? dtype - api : agrmin args : (Tensor x, int axis, bool keepdims, bool flatten, int dtype) output : Tensor infer_meta : func : XXXXInferMeta kernel : func : argmin # arg_max # int64 ???? dtype - api : agrmax args : (Tensor x, int axis, bool keepdims, bool flatten, int dtype) output : Tensor infer_meta : func : XXXXInferMeta kernel : func : argmax # argsort; kernel why input? - api : argsort args : (Tensor x, int axis, bool descending) output : Tensor(out), Tensor(indices) infer_meta : func : XXXXInferMeta kernel : func : argsort # auc; kernel why input? - api : auc args : (Tensor x, Tensor label, Tensor stat_pos, Tenosr stat_neg, string curve, int num_thresholds, int slide_steps) output : Tensor(auc), Tensor(stat_pos_out), Tensor(stat_neg_out) infer_meta : func : AucInferMeta kernel : func : auc # batch_norm - api : batch_norm args : (Tensor x, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, string data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu) output : Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space) infer_meta : func : XXXXInferMeta kernel : func : batch_norm backward: batch_norm_grad # bernoulli - api : bernoulli args : (Tensor x) output : Tensor infer_meta : func : BernoulliInferMeta kernel : func : bernoulli # bilinear_tensor_product ?? optional - api : bilinear_tensor_product args : (Tensor x, Tensor y, Tensor weight, Tensor bias) output : Tensor infer_meta : func : BilinearTensorProductInferMeta kernel : func : bilinear_tensor_product backward : bilinear_tensor_product_grad # bincount ?? optional - api : bincount args : (Tensor x, Tensor weight, int minlength) output : Tensor infer_meta : func : BincountInferMeta kernel : func : bincount # bitwise_and - api : bitwise_and args : (Tensor x, Tensor y) output : Tensor infer_meta : func : BitwiseInferMeta kernel : func : bitwise_and # bitwise_or - api : bitwise_or args : (Tensor x, Tensor y) output : Tensor infer_meta : func : BitwiseInferMeta kernel : func : bitwise_or # bitwise_xor - api : bitwise_xor args : (Tensor x, Tensor y) output : Tensor infer_meta : func : BitwiseInferMeta kernel : func : bitwise_xor # bitwise_not - api : bitwise_not args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : bitwise_not # broadcast_tensors - api : broadcast_tensors args : (Tensor[] x) output : Tensor[] infer_meta : func : BroadcastTensorsInferMeta kernel : func : broadcast_tensors backward : broadcast_tensors_grad # cholesky - api : cholesky args : (Tensor x, bool upper) output : Tensor infer_meta : func : CholeskyInferMeta kernel : func : cholesky backward : cholesky_grad # cholesky_solve - api : cholesky_solve args : (Tensor x, Tensor y, bool upper) output : Tensor infer_meta : func : CholeskySolveInferMeta kernel : func : cholesky_solve backward : cholesky_solve_grad # conv2d # copy # cumsum - api : cumsum args : (Tensor x, int axis, bool flatten, bool exclusive, bool reverse) output : Tensor infer_meta : func : CumsumInferMeta kernel : func : cumsum # depthwise_conv2d # dropout ?? optional, intermediate - api : dropout args : (Tensor x, Tensor seed_tensor, float p, bool is_test, string mode, int seed, bool fix_seed) output : Tensor(out), Tensor(mask) infer_meta : func : DropoutInferMeta kernel : func : dropout # embedding # erf # erfinv # expand_as # expand # eye # flip # gaussian_random # graph_send_recv # histogram # increment # is_empty # isinf # isinf_sr # label_smooth # linspace # log_loss # logical_and # masked_select # multi_dot # multinomial # nll_loss # pad # pixel_shuffle # poisson # psroi_pool # randint_raw # randperm_raw # max # max_raw # reduce_prod # scale_sr # selu # set_value # sgd # shape # shape_sr # shard_index # sigmoid_cross_entropy_with_logits # size # sparse_weight_embedding # tile # top_k # trace # phi_transfer_layout # truncated_gaussian_random # unbind # unfold # uniform_random_raw # uniform_random_raw_sr # viterbi_decode # where_index # yolo_box # sparse_conv3d # dense_to_sparse_coo