- 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 : relu args : (Tensor x) output : Tensor infer_meta : func : UnchangedInferMeta kernel : func : relu 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 inplace : (x -> out) - 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 : 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