api.yaml 4.2 KB
Newer Older
1
- api : add
Z
zyfncg 已提交
2
  args : (Tensor x, Tensor y)
3
  output : Tensor
4
  infer_meta :
5 6
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
7
    func : add
8 9

- api : cast
Z
zyfncg 已提交
10
  args : (Tensor x, DataType out_dtype)
11
  output : Tensor
12
  infer_meta :
13 14 15
    func : CastInferMeta
  kernel :
    func : cast
16
    param : [x, out_dtype]
17 18
    data_type : x

19 20

- api : concat
Z
zyfncg 已提交
21
  args : (Tensor[] x, Scalar axis)
22 23 24
  output : Tensor
  infer_meta :
    func : ConcatInferMeta
25
    param : [x, axis]
26 27 28
  kernel :
    func : concat

29
- api : conj
Z
zyfncg 已提交
30
  args : (Tensor x)
31 32 33 34 35 36
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : conj

37
- api : divide
Z
zyfncg 已提交
38
  args : (Tensor x, Tensor y)
39
  output : Tensor
40
  infer_meta :
41 42
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
43
    func : divide
44 45

- api : dot
Z
zyfncg 已提交
46
  args : (Tensor x, Tensor y)
47
  output : Tensor
48
  infer_meta :
49
    func : DotInferMeta
50
  kernel :
51 52
    func : dot

53
- api : empty
Z
zyfncg 已提交
54
  args : (ScalarArray shape, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU, DataLayout layout=DataLayout::NCHW)
55
  output: Tensor
56
  infer_meta :
57 58
    func : CreateInferMeta
    param : [shape, dtype, layout]
59
  kernel :
60 61 62 63 64
    func : empty
    param : [shape]
    data_type : dtype
    backend : place
    layout : layout
65

66
- api : empty_like
Z
zyfncg 已提交
67
  args : (Tensor x, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED, DataLayout layout = DataLayout::UNDEFINED)
68
  output: Tensor
69
  infer_meta :
70 71
    func : CreateLikeInferMeta
    param : [x, dtype, layout]
72
  kernel :
73 74 75 76 77 78
    func : empty_like
    param : []
    data_type : dtype > x
    backend : place > x
    layout : layout > x

79
- api : flatten
Z
zyfncg 已提交
80
  args : (Tensor x, int start_axis, int stop_axis)
81
  output : Tensor
82
  infer_meta :
83
    func : FlattenInferMeta
84
  kernel :
Y
YuanRisheng 已提交
85
    func : flatten
86 87

- api : full
Z
zyfncg 已提交
88
  args : (ScalarArray shape, Scalar value, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU, DataLayout layout=DataLayout::NCHW)
89
  output: Tensor
90
  infer_meta :
Z
zyfncg 已提交
91
    func : CreateInferMeta
92
    param : [shape, dtype, layout]
93
  kernel :
Y
YuanRisheng 已提交
94
    func : full
95 96 97 98
    param : [shape, value]
    data_type : dtype
    backend : place
    layout : layout
99

100
- api : full_like
Z
zyfncg 已提交
101
  args : (Tensor x, Scalar value, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED, DataLayout layout = DataLayout::UNDEFINED)
102
  output: Tensor
103
  infer_meta :
Z
zyfncg 已提交
104
    func : CreateLikeInferMeta
105
    param : [x, dtype, layout]
106
  kernel :
Y
YuanRisheng 已提交
107
    func : full_like
108
    param : [value]
109 110 111 112 113
    data_type : dtype > x
    backend : place > x
    layout : layout > x

- api : matmul
Z
zyfncg 已提交
114
  args : (Tensor x, Tensor y, bool transpose_x = false, bool transpose_y = false)
115
  output : Tensor
116
  infer_meta :
117
    func : MatmulInferMeta
118
  kernel :
119
    func : matmul
120
  backward : matmul_grad
121 122

- api : mean
Z
zyfncg 已提交
123
  args : (Tensor x, int64_t[] axis={}, bool keep_dim=false)
124
  output : Tensor
125
  infer_meta :
126
    func : ReduceInferMeta
127
  kernel :
Y
YuanRisheng 已提交
128
    func : mean
129 130

- api : multiply
Z
zyfncg 已提交
131
  args : (Tensor x, Tensor y)
132
  output : Tensor
133
  infer_meta :
134 135
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
136
    func : multiply
137 138

- api : ones_like
Z
zyfncg 已提交
139
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED, DataLayout layout=DataLayout::UNDEFINED)
140 141 142 143
  output : Tensor
  invoke : full_like(x, 1, dtype, place, layout)

- api : reshape
Z
zyfncg 已提交
144
  args : (Tensor x, ScalarArray shape)
145
  output : Tensor(out)
146
  infer_meta :
147
    func : ReshapeInferMeta
148
  kernel :
Y
YuanRisheng 已提交
149
    func : reshape
150
  inplace : (x -> out)
151 152

- api : scale
Z
zyfncg 已提交
153
  args : (Tensor x, Scalar scale, float bias, bool bias_after_scale)
154
  output : Tensor
155
  infer_meta :
156 157 158
    func : UnchangedInferMeta
    param : [x]
  kernel :
159
    func : scale, scale_sr
160

161
- api : sign
Z
zyfncg 已提交
162
  args : (Tensor x)
163 164 165 166 167 168
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : sign

169
- api : subtract
Z
zyfncg 已提交
170
  args : (Tensor x, Tensor y)
171
  output : Tensor
172
  infer_meta :
173 174
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
175
    func : subtract
176 177

- api : sum
Z
zyfncg 已提交
178
  args : (Tensor x, int64_t[] axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false)
179
  output : Tensor
180
  infer_meta :
181
    func : SumInferMeta
182
  kernel :
Y
YuanRisheng 已提交
183
    func : sum
184
    param : [x, axis, dtype, keep_dim]
185
    data_type : x
186 187

- api : zeros_like
Z
zyfncg 已提交
188
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED, DataLayout layout=DataLayout::UNDEFINED)
189 190
  output : Tensor
  invoke : full_like(x, 0, dtype, place, layout)