api.yaml 3.9 KB
Newer Older
1 2 3 4 5 6 7
- api : add
  args : (const Tensor& x, const Tensor& y)
  output : Tensor
  infer_meta : 
    func : ElementwiseInferMeta
    param : [x, y, -1]
  kernel :
Y
YuanRisheng 已提交
8
    func : add
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
    param : [x, y, -1]

- api : cast
  args : (const Tensor& x, DataType out_dtype)
  output : Tensor
  infer_meta : 
    func : CastInferMeta
  kernel :
    func : cast
    param : [x, out_dtype, x.dtype()]
    data_type : x

- api : divide
  args : (const Tensor& x, const Tensor& y)
  output : Tensor
  infer_meta : 
    func : ElementwiseInferMeta
    param : [x, y, -1]
  kernel :
Y
YuanRisheng 已提交
28
    func : divide
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
    param : [x, y, -1]

- api : dot
  args : (const Tensor& x, const Tensor& y)
  output : Tensor
  infer_meta : 
    func : DotInferMeta
  kernel : 
    func : dot

- api : flatten
  args : (const Tensor& x, int start_axis, int stop_axis)
  output : Tensor
  infer_meta : 
    func : FlattenInferMeta
  kernel : 
Y
YuanRisheng 已提交
45
    func : flatten
46 47 48 49 50 51 52 53

- api : full
  args : (const ScalarArray& shape, const Scalar& value, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU, DataLayout layout=DataLayout::NCHW)
  output: Tensor
  infer_meta : 
    func : FullInferMeta
    param : [shape, dtype, layout]
  kernel : 
Y
YuanRisheng 已提交
54
    func : full
55 56 57 58 59 60 61 62 63 64 65 66
    param : [shape, value]
    data_type : dtype
    backend : place
    layout : layout
  
- api : full_like
  args : (const Tensor& x, const Scalar& value, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED, DataLayout layout = DataLayout::UNDEFINED)
  output: Tensor
  infer_meta : 
    func : FullLikeInferMeta
    param : [x, dtype, layout]
  kernel : 
Y
YuanRisheng 已提交
67
    func : full_like
68
    param : [value]
69 70 71 72 73 74 75 76 77 78
    data_type : dtype > x
    backend : place > x
    layout : layout > x

- api : matmul
  args : (const Tensor& x, const Tensor& y, bool transpose_x = false, bool transpose_y = false)
  output : Tensor
  infer_meta : 
    func : MatmulInferMeta
  kernel : 
79
    func : matmul
80 81

- api : mean
82
  args : (const Tensor& x, const std::vector<int64_t>& axis={}, bool keep_dim=false)
83 84 85
  output : Tensor
  infer_meta : 
    func : ReduceInferMeta
86
    param: [x, axis, keep_dim]
87
  kernel : 
Y
YuanRisheng 已提交
88
    func : mean
89
    param : [x, axis, keep_dim, false]
90 91 92 93 94 95 96 97

- api : multiply
  args : (const Tensor& x, const Tensor& y)
  output : Tensor
  infer_meta : 
    func : ElementwiseInferMeta
    param : [x, y, -1]
  kernel :
Y
YuanRisheng 已提交
98
    func : multiply
99 100 101 102 103 104 105 106
    param : [x, y, -1]

- api : ones_like
  args : (const Tensor& x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED, DataLayout layout=DataLayout::UNDEFINED)
  output : Tensor
  invoke : full_like(x, 1, dtype, place, layout)

- api : reshape
107
  args : (const Tensor& x, const ScalarArray& shape)
108 109
  output : Tensor
  infer_meta : 
110
    func : ReshapeInferMeta
111
  kernel : 
Y
YuanRisheng 已提交
112
    func : reshape
113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129

- api : scale
  args : (const Tensor& x, const Scalar& scale, float bias, bool bias_after_scale)
  output : Tensor
  infer_meta : 
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : scale

- api : subtract
  args : (const Tensor& x, const Tensor& y)
  output : Tensor
  infer_meta : 
    func : ElementwiseInferMeta
    param : [x, y, -1]
  kernel :
Y
YuanRisheng 已提交
130
    func : subtract
131 132 133
    param : [x, y, -1]

- api : sum
134
  args : (const Tensor& x, const std::vector<int64_t>& axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false)
135 136 137
  output : Tensor
  infer_meta : 
    func : ReduceInferMeta
138
    param: [x, axis, keep_dim, dtype]
139
  kernel : 
Y
YuanRisheng 已提交
140
    func : sum
141 142
    param : [x, axis, keep_dim, false, DataType::UNDEFINED]
    data_type : x
143 144 145 146 147 148 149 150 151 152 153 154 155 156

- api : zeros_like
  args : (const Tensor& x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED, DataLayout layout=DataLayout::UNDEFINED)
  output : Tensor
  invoke : full_like(x, 0, dtype, place, layout)

# - api : full_like
#   args : (const Tensor& x, const Scalar& value, DataType dtype, Backend place)->Tensor
#   output: {Tensor : dtype}
#   kernel : fill_any_like
#   T : [dtype, x]
#   backend : [place, x]
#   layout : []
#   InferMeta : UnchangedInferMeta(x)