- api : add args : (Tensor x, Tensor y) output : Tensor infer_meta : func : ElementwiseInferMeta kernel : func : add # backward : add_grad - 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, Place place, bool blocking) output : Tensor invoke : copy_to_impl(x, place, 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, Place place=CPUPlace()) 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, Place place = {}) 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, Place place=CPUPlace()) 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, Place place = {}) 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, Place place={}) output : Tensor invoke : full_like(x, 1, dtype, place) - api : pool2d args : (Tensor x, int[] kernel_size, int[] strides, int[] paddings, bool ceil_mode, bool exclusive, str data_format, str pooling_type, bool global_pooling, bool adaptive, str padding_algorithm) output : Tensor(out) infer_meta : func : PoolInferMeta kernel: func : pool2d - api : reshape args : (Tensor x, ScalarArray shape) output : Tensor(out) infer_meta : func : ReshapeInferMeta kernel : func : reshape inplace : (x -> out) - api : relu args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : relu inplace : (x -> out) backward: relu_grad - 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 inplace : (x -> out) - api : sign args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : sign - api : softmax args : (Tensor x, int axis) output : Tensor infer_meta : func : SoftmaxInferMeta kernel : func : sotfmax - 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, Place place = {}) output : Tensor invoke : full_like(x, 0, dtype, place) - api : one_hot args : (Tensor x, Scalar num_classes) output : Tensor infer_meta : func : OneHotInferMeta kernel : func : one_hot - 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 - api : gumbel_softmax args : (Tensor x, float temperature, bool hard, int axis) output : Tensor infer_meta : func : GumbelSoftmaxInferMeta kernel : func : gumbel_softmax # backward : gumbel_softmax_grad - api : diag args : (Tensor x, int offset, float padding_value) output : Tensor infer_meta : func : DiagInferMeta kernel : func : diag # - api : pixel_shuffle # args : (Tensor x, int upscale_factor, const std::string& data_format) # output : Tensor # infer_meta : # func : PixelShuffleInferMeta # kernel : # func : pixel_shuffle - api : transpose args : (Tensor x, int[] axis) output : Tensor infer_meta : func : TransposeInferMeta kernel : func : transpose backward : transpose_grad - api : lerp args : (Tensor x, Tensor y, Tensor weight) output : Tensor infer_meta : func : LerpInferMeta kernel : func : lerp # backward : lerp_grad - api : scatter args : (Tensor x, Tensor index, Tensor updates, bool overwrite) output : Tensor infer_meta : func : ScatterInferMeta dtype : x kernel : func : scatter backward : scatter_grad - api : scatter_nd_add args : (Tensor x, Tensor index, Tensor updates) output : Tensor infer_meta : func : ScatterNdAddInferMeta dtype : x kernel : func : scatter_nd_add backward : scatter_nd_add_grad - api : addmm args : (Tensor input, Tensor x, Tensor y, float alpha, float beta) output : Tensor infer_meta : func : AddmmInferMeta kernel : func : addmm backward : addmm_grad - api : adadelta args : (Tensor param, Tensor grad, Tensor avg_squared_grad, Tensor avg_squared_update, float rho, float epsilon) output : Tensor(param_out), Tensor(moment_out), Tensor(inf_norm_out) infer_meta : func : AdadeltaInferMeta kernel : func : adadelta - api : adamax args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment, Tensor inf_norm, Tensor beta1_pow, float beta1, float beta2, float epsilon) output : Tensor(param_out), Tensor(avg_squared_grad_out), Tensor(avg_squared_update_out) infer_meta : func : AdamaxInferMeta kernel : func : adamax - api : where args : (Tensor condition, Tensor x, Tensor y) output : Tensor infer_meta : func : WhereInferMeta kernel : func : where backward : where_grad # BilinearTensorProductInferMeta # BroadcastTensorsInferMeta - api : less_than args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : less_than - api : less_equal args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : less_equal - api : greater args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : greater - api : greater_equal args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : greater_equal - api : equal args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : equal - api : not_equal args : (Tensor x, Tensor y, int axis = -1) output : Tensor infer_meta : func : CompareInferMeta kernel : func : not_equal # - api : equal_all # args : (Tensor x, Tensor y) # output : Tensor # infer_meta : # func : CompareAllInferMeta # kernel : # func : equal_all - api : huber_loss args : (Tensor input, Tensor label, float delta) output : Tensor(out), Tensor(residual) infer_meta : func : HuberLossInferMeta kernel : func : huber_loss # backward : huber_loss_grad - api : triangular_solve args : (Tensor x, Tensor y, bool upper, bool tranpose, bool unitriangular) output : Tensor infer_meta : func : TriangularSolveInferMeta kernel : func : triangular_solve # backward : triangular_solve_grad - api : index_sample args : (Tensor x, Tensor index) output : Tensor infer_meta : func : IndexSampleInferMeta kernel : func : index_sample data_type : x backward : index_sample_grad - api : cross args : (Tensor x, Tensor y, int axis = 9) output : Tensor infer_meta : func : CrossInferMeta kernel : func : cross backward : cross_grad - api : atan2 args : (Tensor x, Tensor y) output : Tensor infer_meta : func : Atan2InferMeta kernel : func : atan2 backward : atan2_grad - api : bce_loss args : (Tensor input, Tensor label) output : Tensor infer_meta : func : BCELossInferMeta kernel : func : bce_loss backward : bce_loss_grad - api : dist args : (Tensor x, Tensor y, float p) output : Tensor infer_meta : func : DistInferMeta kernel : func : dist # backward : dist_grad - api : gather_nd args : (Tensor x, Tensor index) output : Tensor infer_meta : func : GatherNdInferMeta kernel : func : gather_nd data_type : x backward : gather_nd_grad - api : gather_tree args : (Tensor ids, Tensor parents) output : Tensor infer_meta : func : GatherTreeMeta kernel : func : gather_tree - api : mv args : (Tensor x, Tensor vec) output : Tensor infer_meta : func : MvInferMeta kernel : func : mv backward : mv_grad # =================================== sep0 # =================================== sep1 # =================================== sep2 # =================================== sep3