api.yaml 24.7 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 38 39 40 41
- api : copy_to
  args : (Tensor x, Backend backend, bool blocking)
  output : Tensor
  invoke : copy_to_impl(x, backend, blocking)

42
- api : divide
Z
zyfncg 已提交
43
  args : (Tensor x, Tensor y)
44
  output : Tensor
45
  infer_meta :
46 47
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
48
    func : divide
49 50

- api : dot
Z
zyfncg 已提交
51
  args : (Tensor x, Tensor y)
52
  output : Tensor
53
  infer_meta :
54
    func : DotInferMeta
55
  kernel :
56 57
    func : dot

58
- api : empty
59
  args : (ScalarArray shape, DataType dtype=DataType::FLOAT32, Backend place=Backend::CPU)
60
  output: Tensor
61
  infer_meta :
62
    func : CreateInferMeta
63
    param : [shape, dtype]
64
  kernel :
65
    func : empty
66
    param : [shape, dtype]
67 68
    data_type : dtype
    backend : place
69

70
- api : empty_like
71
  args : (Tensor x, DataType dtype = DataType::UNDEFINED, Backend place = Backend::UNDEFINED)
72
  output: Tensor
73
  infer_meta :
74
    func : CreateLikeInferMeta
75
    param : [x, dtype]
76
  kernel :
77
    func : empty_like
78
    param : [x, dtype]
79 80 81
    data_type : dtype > x
    backend : place > x

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

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

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

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

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

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

- api : ones_like
140
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED)
141
  output : Tensor
142
  invoke : full_like(x, 1, dtype, place)
143

F
From00 已提交
144 145 146 147 148 149 150 151
- 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

152
- api : reshape
Z
zyfncg 已提交
153
  args : (Tensor x, ScalarArray shape)
154
  output : Tensor(out)
155
  infer_meta :
156
    func : ReshapeInferMeta
157
  kernel :
Y
YuanRisheng 已提交
158
    func : reshape
159
  inplace : (x -> out)
160

Y
YuanRisheng 已提交
161 162 163 164 165 166 167 168
- api : relu
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : relu
  inplace : (x -> out)
P
phlrain 已提交
169
  backward : relu_grad
Y
YuanRisheng 已提交
170

171
- api : scale
Z
zyfncg 已提交
172
  args : (Tensor x, Scalar scale, float bias, bool bias_after_scale)
173
  output : Tensor
174
  infer_meta :
175 176 177
    func : UnchangedInferMeta
    param : [x]
  kernel :
178
    func : scale, scale_sr
179
  inplace : (x -> out)
180

181
- api : sign
Z
zyfncg 已提交
182
  args : (Tensor x)
183 184 185 186 187 188
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : sign

189 190 191 192 193 194
- api : softmax
  args : (Tensor x, int axis)
  output : Tensor
  infer_meta :
    func : SoftmaxInferMeta
  kernel :
P
phlrain 已提交
195 196
    func : softmax
  backward : softmax_grad
197

198 199 200 201 202
- api : split
  args : (Tensor x, ScalarArray num_or_sections, Scalar axis)
  output : Tensor[]
  invoke : split_impl(x, num_or_sections, axis)

203
- api : subtract
Z
zyfncg 已提交
204
  args : (Tensor x, Tensor y)
205
  output : Tensor
206
  infer_meta :
207 208
    func : ElementwiseInferMeta
  kernel :
Y
YuanRisheng 已提交
209
    func : subtract
210 211

- api : sum
212
  args : (Tensor x, int64[] axis={}, DataType dtype=DataType::UNDEFINED, bool keep_dim=false)
213
  output : Tensor
214
  infer_meta :
215
    func : SumInferMeta
216
  kernel :
Y
YuanRisheng 已提交
217
    func : sum
218
    data_type : x
219 220

- api : zeros_like
221
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Backend place=Backend::UNDEFINED)
222
  output : Tensor
223
  invoke : full_like(x, 0, dtype, place)
H
hong 已提交
224

H
hong 已提交
225 226 227 228 229 230 231 232 233

- api : one_hot
  args : (Tensor x, Scalar num_classes)
  output : Tensor
  infer_meta :
    func : OneHotInferMeta
  kernel :
    func : one_hot
    
H
hong 已提交
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
- 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
P
phlrain 已提交
279

P
phlrain 已提交
280 281 282 283 284 285 286 287 288
# # maxout
# - api : maxout
#   args : (Tensor x, int groups, int axis)
#   output : Tensor
#   infer_meta :
#     func : MaxoutInferMeta
#   kernel :
#     func : maxout
#   backward : maxout_grad
P
phlrain 已提交
289

P
update  
phlrain 已提交
290 291 292 293 294 295 296 297 298
# put_along_axis
- api : put_along_axis
  args : (Tensor x, Tensor index, Tensor value, int axis, str reduce)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [index]
  kernel :
    func : put_along_axis
P
phlrain 已提交
299
    dtype : x
P
update  
phlrain 已提交
300
  backward : put_along_axis_grad
P
phlrain 已提交
301 302


P
update  
phlrain 已提交
303 304 305 306 307 308 309 310 311
# take_along_axis
- api : take_along_axis
  args : (Tensor x, Tensor index, int axis)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [index]
  kernel :
    func : take_along_axis
P
phlrain 已提交
312
    dtype : x
P
update  
phlrain 已提交
313
  backward : take_along_axis_grad
P
phlrain 已提交
314

P
update  
phlrain 已提交
315 316 317 318 319 320 321 322 323 324
# matrix_power
- api : matrix_power
  args : (Tensor x, int n)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : matrix_power
  backward : matrix_power_grad
P
phlrain 已提交
325

P
update  
phlrain 已提交
326 327 328 329 330 331 332 333 334
# eigh
- api : eigh
  args : (Tensor x, str uplo)
  output : Tensor(out_w), Tensor(out_v)
  infer_meta :
    func : EighInferMeta
  kernel :
    func : eigh
  backward : eigh_grad
P
phlrain 已提交
335

P
update  
phlrain 已提交
336 337 338 339 340 341 342 343 344
# segment_pool
- api : segment_pool
  args : (Tensor x, Tensor segment_ids, str pooltype)
  output : Tensor(out), Tensor(summed_ids)
  infer_meta :
    func : SegmentPoolInferMeta
  kernel :
    func : segment_pool
  backward : segment_pool_grad
P
phlrain 已提交
345
  
P
phlrain 已提交
346

P
phlrain 已提交
347 348 349 350 351 352 353 354 355
# # accuracy
# - api : accuracy
#   args : (Tensor x, Tensor indices, Tensor label)
#   output : Tensor(accuracy), Tensor(correct), Tensor(total)
#   infer_meta :
#     func : AccuracyInferMeta
#   kernel :
#     func : accuracy
#     dtype : x
P
phlrain 已提交
356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469

# 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

P
phlrain 已提交
470 471 472 473 474 475 476 477 478
# sigmoid
- api : sigmoid
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : sigmoid
  backward : sigmoid_grad
P
phlrain 已提交
479

P
phlrain 已提交
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
# tan
- api : tan
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : tan
  backward : tan_grad

# tanh_shrink
- api : tanh_shrink
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : tanh_shrink
  backward : tanh_shrink_grad

# silu
- api : silu
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : silu
  backward : silu_grad

# logsigmoid
- api : logsigmoid
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : logsigmoid
  backward : logsigmoid_grad

# leaky_relu
- api : leaky_relu
  args : (Tensor x, float alpha)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : leaky_relu
  backward : leaky_relu_grad

# thresholded_relu
- api : thresholded_relu
  args : (Tensor x, float threshold)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : thresholded_relu
  backward : thresholded_relu_grad


# soft_shrink
- api : soft_shrink
  args : (Tensor x, float lambda)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : soft_shrink
  backward : soft_shrink_grad

# hard_shrink
- api : hard_shrink
  args : (Tensor x, float threshold)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : hard_shrink
  backward : hard_shrink_grad


# elu
- api : elu
  args : (Tensor x, float alpha)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : elu
  backward : elu_grad

# brelu
- api : brelu
  args : (Tensor x, float t_min, float t_max)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : brelu
  backward : brelu_grad

# hard_sigmoid
- api : hard_sigmoid
  args : (Tensor x, float slope, float offset)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : hard_sigmoid
  backward : hard_sigmoid_grad


P
phlrain 已提交
600
# arg_min  # int64 ???? dtype
P
phlrain 已提交
601
- api : argmin
P
update  
phlrain 已提交
602
  args : (Tensor x, int64 axis, bool keepdims, bool flatten, int dtype)
P
phlrain 已提交
603 604
  output : Tensor
  infer_meta :
P
phlrain 已提交
605
    func : ArgMinMaxInferMeta
P
phlrain 已提交
606
  kernel :
P
update  
phlrain 已提交
607
    func : arg_min
P
phlrain 已提交
608 609

# arg_max  # int64 ???? dtype
P
phlrain 已提交
610
- api : argmax
P
update  
phlrain 已提交
611
  args : (Tensor x, int64 axis, bool keepdims, bool flatten, int dtype)
P
phlrain 已提交
612 613
  output : Tensor
  infer_meta :
P
phlrain 已提交
614
    func : ArgMinMaxInferMeta
P
phlrain 已提交
615
  kernel :
P
update  
phlrain 已提交
616
    func : arg_max
P
phlrain 已提交
617

P
update  
phlrain 已提交
618
# argsort
P
phlrain 已提交
619 620
- api : argsort
  args : (Tensor x, int axis, bool descending)
P
add  
phlrain 已提交
621
  output : Tensor(out), Tensor(indices)
P
phlrain 已提交
622
  infer_meta :
P
update  
phlrain 已提交
623
    func : ArgsortInferMeta 
P
phlrain 已提交
624 625
  kernel :
    func : argsort
P
update  
phlrain 已提交
626
  backward : argsort_grad
P
phlrain 已提交
627

P
update  
phlrain 已提交
628 629 630 631 632 633 634 635
# auc
- api : auc
  args : (Tensor x, Tensor label, Tensor stat_pos, Tensor stat_neg, str 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
P
phlrain 已提交
636

P
phlrain 已提交
637 638
# # batch_norm
# - api : batch_norm
P
update  
phlrain 已提交
639
#   args : (Tensor x, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, str data_layout, bool is_test, bool use_global_stats, bool trainable_statistics, bool fuse_with_relu)
P
phlrain 已提交
640 641 642 643 644 645
#   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
P
add  
phlrain 已提交
646

P
update  
phlrain 已提交
647 648 649 650 651 652 653 654
# bernoulli
- api : bernoulli
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : bernoulli
P
add  
phlrain 已提交
655

P
phlrain 已提交
656 657 658 659 660 661 662 663 664
# # 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
P
update  
phlrain 已提交
665 666 667 668 669 670 671 672 673 674 675
#   optional : bias

# bincount
- api : bincount
  args : (Tensor x, Tensor weight, int minlength)
  output : Tensor
  infer_meta :
    func : BincountInferMeta
  kernel :
    func : bincount
  optional : weight
P
add  
phlrain 已提交
676 677 678



P
phlrain 已提交
679
# bitwise_and
P
phlrain 已提交
680 681 682 683 684 685 686
- api : bitwise_and
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : bitwise_and
P
add  
phlrain 已提交
687

P
phlrain 已提交
688 689 690 691 692 693 694 695
# bitwise_or
- api : bitwise_or
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : bitwise_or
P
add  
phlrain 已提交
696 697


P
phlrain 已提交
698 699 700 701 702 703 704 705
# bitwise_xor
- api : bitwise_xor
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : bitwise_xor
P
add  
phlrain 已提交
706 707 708 709 710 711 712 713 714 715

# bitwise_not
- api : bitwise_not
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : bitwise_not

P
phlrain 已提交
716
# broadcast_tensors
P
phlrain 已提交
717 718 719 720 721 722 723 724
# - api : broadcast_tensors
#   args : (Tensor[] x)
#   output : Tensor[]
#   infer_meta :
#     func : BroadcastTensorsInferMeta
#   kernel :
#     func : broadcast_tensors
#   backward : broadcast_tensors_grad
P
add  
phlrain 已提交
725

P
phlrain 已提交
726
# cholesky
P
add  
phlrain 已提交
727 728 729 730 731 732 733 734 735
- api : cholesky
  args : (Tensor x, bool upper)
  output : Tensor
  infer_meta :
    func : CholeskyInferMeta
  kernel :
    func : cholesky
  backward : cholesky_grad
  
P
phlrain 已提交
736
# cholesky_solve
P
add  
phlrain 已提交
737 738 739 740 741 742 743 744 745
- api : cholesky_solve
  args : (Tensor x, Tensor y, bool upper)
  output : Tensor
  infer_meta :
    func : CholeskySolveInferMeta
  kernel :
    func : cholesky_solve
  backward : cholesky_solve_grad

P
phlrain 已提交
746 747
# # conv2d
# # copy
P
add  
phlrain 已提交
748 749


P
phlrain 已提交
750
# cumsum
P
add  
phlrain 已提交
751 752 753 754 755 756 757 758
- api : cumsum
  args : (Tensor x, int axis, bool flatten, bool exclusive, bool reverse)
  output : Tensor
  infer_meta :
    func : CumsumInferMeta
  kernel :
    func : cumsum

P
phlrain 已提交
759 760 761
# # depthwise_conv2d
# # dropout ?? optional, intermediate
# - api : dropout
P
update  
phlrain 已提交
762
#   args : (Tensor x, Tensor seed_tensor, float p, bool is_test, str mode, int seed, bool fix_seed)
P
phlrain 已提交
763 764 765 766 767
#   output : Tensor(out), Tensor(mask)
#   infer_meta :
#     func : DropoutInferMeta
#   kernel :
#     func : dropout
P
add  
phlrain 已提交
768

P
phlrain 已提交
769
# # embedding
P
update  
phlrain 已提交
770

P
update  
phlrain 已提交
771 772 773 774 775 776 777 778 779
# erf
- api : erf
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : erf
  backward : erf_grad
P
update  
phlrain 已提交
780

P
update  
phlrain 已提交
781 782 783 784 785 786 787 788 789
# erfinv
- api : erfinv
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : erfinv
  backward : erfinv_grad
P
update  
phlrain 已提交
790

P
update  
phlrain 已提交
791
# expand_as
P
phlrain 已提交
792 793 794 795 796 797 798 799
# - api : expand_as
#   args : (Tensor x, Tensor y, int[] shape)
#   output : Tensor
#   infer_meta :
#     func : ExpandAsInferMeta
#   kernel :
#     func : expand_as
#   backward : expand_as_grad
P
update  
phlrain 已提交
800
#   optional : y
P
update  
phlrain 已提交
801 802


P
phlrain 已提交
803 804 805 806 807 808 809 810 811
# # expand
# - api : expand
#   args : (Tensor x, ScalarArray shape)
#   output : Tensor
#   infer_meta :
#     func : ExpandInferMeta
#   kernel :
#     func : expand
#   backward : expand_grad
P
update  
phlrain 已提交
812

P
phlrain 已提交
813
# eye
P
update  
phlrain 已提交
814 815 816 817 818 819 820
# - api : eye
#   args : (int64 num_rows, int64 num_colums, DataType dtype = DataType::FLOAT32)
#   output : Tensor
#   infer_meta :
#     func : EyeInferMeta
#   kernel :
#     func : eye
P
update  
phlrain 已提交
821

P
phlrain 已提交
822 823 824 825 826 827 828 829
# # flip
# - api : flip
#   args : (Tensor x, int[] axis)
#   output : Tensor
#   infer_meta :
#     func : FlipInferMeta
#   kernel :
#     func : flip
P
update  
phlrain 已提交
830

P
phlrain 已提交
831
# gaussian_random
P
phlrain 已提交
832
# - api : gaussian_random
P
update  
phlrain 已提交
833
#   args : (ScalarArray shape, float mean, float std, int seed, DataType dtype=DataType::FLOAT32)
P
phlrain 已提交
834 835
#   output : Tensor
#   infer_meta :
P
update  
phlrain 已提交
836 837
#     func : CreateInferMeta
#     param : [shape, dtype]
P
phlrain 已提交
838 839
#   kernel :
#     func : gaussian_random
P
update  
phlrain 已提交
840
#     data_type : dtype
P
update  
phlrain 已提交
841

P
phlrain 已提交
842 843
# # graph_send_recv
# - api : graph_send_recv
P
update  
phlrain 已提交
844
#   args : (Tensor x, Tensor src_index, Tensor dst_index, str pool_type)
P
phlrain 已提交
845 846 847 848 849 850
#   output : Tensor(out), Tensor(dst_count)
#   infer_meta :
#     func : GraphSendRecvInferMeta
#   kernel :
#     func : graph_send_recv
#   backward : graph_send_recv_grad
P
update  
phlrain 已提交
851

P
phlrain 已提交
852 853
# # histogram int64 ???
# - api : histogram
P
update  
phlrain 已提交
854
#   args : (Tensor x, int64 bins, int min, int max)
P
phlrain 已提交
855 856 857 858 859
#   output : Tensor
#   infer_meta :
#     func : HistogramInferMeta
#   kernel :
#     func : histogram
P
update  
phlrain 已提交
860

P
update  
phlrain 已提交
861 862 863 864 865 866 867 868
# increment
- api : increment
  args : (Tensor x, float value)
  output : Tensor
  infer_meta :
    func : IncrementInferMeta
  kernel :
    func : increment
P
update  
phlrain 已提交
869

P
update  
phlrain 已提交
870 871 872 873 874 875 876 877
# is_empty
- api : is_empty
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : IsEmptyInferMeta
  kernel :
    func : is_empty
P
update  
phlrain 已提交
878

P
phlrain 已提交
879 880 881 882 883 884 885 886
# # isinf selected rows??? involk
# - api : isinf
#   args : (Tensor x)
#   output : Tensor
#   infer_meta :
#     func : IsfiniteInferMeta
#   kernel :
#     func : isinf
P
update  
phlrain 已提交
887

P
update  
phlrain 已提交
888
# # isnan selected rows??? involk
P
phlrain 已提交
889 890 891 892 893 894 895
# - api : isnan
#   args : (Tensor x)
#   output : Tensor
#   infer_meta :
#     func : IsfiniteInferMeta
#   kernel :
#     func : isnan
P
update  
phlrain 已提交
896

P
update  
phlrain 已提交
897
# # isfinite selected rows??? involk
P
phlrain 已提交
898 899 900 901 902 903 904
# - api : isfinite
#   args : (Tensor x)
#   output : Tensor
#   infer_meta :
#     func : IsfiniteInferMeta
#   kernel :
#     func : isfinite
P
update  
phlrain 已提交
905

P
update  
phlrain 已提交
906
# label_smooth  ?? optional
P
phlrain 已提交
907 908 909 910
# - api : label_smooth
#   args : (Tensor label, Tensor prior_dist, float epsilon)
#   output : Tensor
#   infer_meta :
P
update  
phlrain 已提交
911 912
#     func : UnchangedInferMeta
#     param : [label]
P
phlrain 已提交
913 914 915
#   kernel :
#     func : label_smooth
#   backward : label_smooth_grad
P
update  
phlrain 已提交
916
#   optional : prior_dist
P
update  
phlrain 已提交
917

P
update  
phlrain 已提交
918
# linspace ???? start stop number,应该是sclar?
P
phlrain 已提交
919
# - api : linspace
P
update  
phlrain 已提交
920
#   args : (Tensor start, Tensor stop, Tensor number, DataType dtype=DataType::FLOAT32)
P
phlrain 已提交
921 922 923 924 925
#   output : Tensor
#   infer_meta :
#     func : LinspaceInferMeta
#   kernel :
#     func : linspace
P
update  
phlrain 已提交
926

P
update  
phlrain 已提交
927 928 929 930 931 932 933 934 935 936

# log_loss
- api : log_loss
  args : (Tensor input, Tensor label, float epsilon)
  output : Tensor
  infer_meta :
    func : LogLossInferMeta
  kernel :
    func : log_loss
  backward : log_loss_grad
P
update  
phlrain 已提交
937

P
phlrain 已提交
938 939 940 941 942 943 944 945
# # logical_and
# - api : logical_and
#   args : (Tensor x, Tensor y)
#   output : Tensor
#   infer_meta :
#     func : LogicalInferMeta
#   kernel :
#     func : logical_and
P
update  
phlrain 已提交
946

P
phlrain 已提交
947 948 949 950 951 952 953 954
# # logical_or
# - api : logical_or
#   args : (Tensor x, Tensor y)
#   output : Tensor
#   infer_meta :
#     func : LogicalInferMeta
#   kernel :
#     func : logical_or
P
phlrain 已提交
955 956


P
phlrain 已提交
957 958 959 960 961 962 963 964
# # logical_xor
# - api : logical_xor
#   args : (Tensor x, Tensor y)
#   output : Tensor
#   infer_meta :
#     func : LogicalInferMeta
#   kernel :
#     func : logical_xor
P
phlrain 已提交
965

P
update  
phlrain 已提交
966 967 968 969 970 971 972 973
# logical_not
- api : logical_not
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : logical_not
P
update  
phlrain 已提交
974

P
phlrain 已提交
975 976 977 978 979 980 981 982 983
# # masked_select
# - api : masked_select
#   args : (Tensor x, Tensor mask)
#   output : Tensor
#   infer_meta :
#     func : MaksedSelectInferMeta
#   kernel :
#     func : masked_select
#   backward : masked_select_grad
P
phlrain 已提交
984

P
phlrain 已提交
985 986 987 988 989 990 991 992 993
# # multi_dot
# - api : multi_dot
#   args : (Tensor[] x)
#   output : Tensor
#   infer_meta :
#     func : MultiDotInferMeta
#   kernel :
#     func : multi_dot
#   backward : multi_dot_grad
P
phlrain 已提交
994

P
update  
phlrain 已提交
995 996 997 998 999 1000 1001 1002
# multinomial
- api : multinomial
  args : (Tensor x, int num_samples, bool replacement)
  output : Tensor
  infer_meta :
    func : MultinomialInferMeta
  kernel :
    func : multinomial
P
phlrain 已提交
1003

P
update  
phlrain 已提交
1004
# nll_loss  ?? optional
P
phlrain 已提交
1005
# - api : nll_loss
P
update  
phlrain 已提交
1006
#   args : (Tensor x, Tensor label, Tensor weight, int64 ignore_index, str reduction)
P
phlrain 已提交
1007 1008
#   output : Tensor(out), Tensor(total_weight)
#   infer_meta :
P
update  
phlrain 已提交
1009
#     func : NllLossRawInferMeta
P
phlrain 已提交
1010 1011 1012
#   kernel :
#     func : nll_loss
#   backward : nll_loss_grad
P
update  
phlrain 已提交
1013
#   optional : weight
P
phlrain 已提交
1014

P
phlrain 已提交
1015 1016 1017 1018 1019 1020 1021 1022 1023
# # pad
# - api : pad
#   args : (Tensor x, int[] paddings, float pad_value)
#   output : Tensor
#   infer_meta :
#     func : PadInferMeta
#   kernel :
#     func : pad
#   backward : pad_grad
P
phlrain 已提交
1024

P
phlrain 已提交
1025 1026
# # pixel_shuffle
# - api : pixel_shuffle
P
update  
phlrain 已提交
1027
#   args : (Tensor x, int upscale_factor, str data_format)
P
phlrain 已提交
1028 1029 1030 1031 1032 1033
#   output : Tensor
#   infer_meta :
#     func : PixelShuffleInferMeta
#   kernel :
#     func : pixel_shuffle
#   backward : pixel_shuffle_grad
P
phlrain 已提交
1034

P
phlrain 已提交
1035 1036 1037 1038 1039 1040 1041 1042 1043
# # poisson
# - api : poisson
#   args : (Tensor x)
#   output : Tensor
#   infer_meta :
#     func : PoissonInferMeta
#   kernel :
#     func : poisson
#   backward : poisson_grad
P
phlrain 已提交
1044

P
update  
phlrain 已提交
1045
# psroi_pool  ?? optional
P
phlrain 已提交
1046 1047 1048 1049 1050 1051 1052 1053
# - api : psroi_pool
#   args : (Tensor x, Tensor rois, Tensor rois_num, int pooled_weight, int pooled_width, int output_channels, float spatial_scale )
#   output : Tensor
#   infer_meta :
#     func : PsroiPoolInferMeta
#   kernel :
#     func : psroi_pool
#   backward : psroi_pool_grad
P
update  
phlrain 已提交
1054
#   optional : rois_num
P
phlrain 已提交
1055

P
phlrain 已提交
1056 1057 1058 1059 1060 1061 1062 1063
# # randint_raw
# - api : randint
#   args : (int low, int high, ScalarArray shape, DataType dtype)
#   output : Tensor
#   infer_meta :
#     func : RandintInferMeta
#   kernel :
#     func : randint
P
phlrain 已提交
1064

P
phlrain 已提交
1065 1066 1067 1068 1069 1070 1071 1072
# # randperm_raw
# - api : randperm
#   args : (int n, DataType dtype)
#   output : Tensor
#   infer_meta :
#     func : RandpermInferMeta
#   kernel :
#     func : randperm
P
phlrain 已提交
1073

P
phlrain 已提交
1074 1075 1076 1077 1078 1079 1080 1081
# # max
# - api : max
#   args : (Tensor x, int64_t[] dims, bool keep_dim)
#   output : Tensor
#   infer_meta :
#     func : MaxInferMeta
#   kernel :
#     func : max
P
phlrain 已提交
1082

P
phlrain 已提交
1083 1084 1085 1086 1087 1088 1089 1090
# # reduce_prod
# - api : reduce_prod
#   args : (Tensor x, int64_t[] dims, bool keep_dim, bool reduce_all)
#   output : Tensor
#   infer_meta :
#     func : ReduceProdInferMeta
#   kernel :
#     func : reduce_prod
P
phlrain 已提交
1091

P
update  
phlrain 已提交
1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
# selu
- api : selu
  args : (Tensor x, float scale, float alpha)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : selu
  backward : selu_grad
P
phlrain 已提交
1102

P
phlrain 已提交
1103
# # set_value None api
P
phlrain 已提交
1104

P
phlrain 已提交
1105 1106
# # sgd # need invoke
# # shape ??? selcted rows
P
phlrain 已提交
1107

P
update  
phlrain 已提交
1108 1109 1110 1111 1112 1113 1114 1115
# shard_index
- api : shard_index
  args : (Tensor in, int index_num, int nshards, int shard_id, int ignore_value)
  output : Tensor
  infer_meta :
    func : ShardIndexInferMeta
  kernel :
    func : shard_index
P
phlrain 已提交
1116

P
update  
phlrain 已提交
1117 1118 1119 1120 1121 1122 1123 1124 1125
# sigmoid_cross_entropy_with_logits
- api : sigmoid_cross_entropy_with_logits
  args : (Tensor x, Tensor label, bool normalize, int ignore_index)
  output : Tensor
  infer_meta :
    func : SigmoidCrossEntropyWithLogitsInferMeta
  kernel :
    func : sigmoid_cross_entropy_with_logits
  backward : sigmoid_cross_entropy_with_logits_grad
P
phlrain 已提交
1126

P
update  
phlrain 已提交
1127 1128 1129 1130 1131 1132 1133 1134
# size
- api : size
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : SizeInferMeta
  kernel :
    func : size
P
phlrain 已提交
1135

P
update  
phlrain 已提交
1136 1137 1138 1139 1140 1141 1142 1143 1144
# tile
- api : tile
  args : (Tensor x, ScalarArray repeat_times)
  output : Tensor
  infer_meta :
    func : TileInferMeta
  kernel :
    func : tile
  backward : tile_grad
P
phlrain 已提交
1145

P
phlrain 已提交
1146 1147 1148 1149 1150 1151 1152 1153 1154
# # top_k
# - api : top_k
#   args : (Tensor x, Scalar k, int axis, bool largest, bool sorted)
#   output : Tensor(out), Tensor(indices)
#   infer_meta :
#     func : TopkInferMeta
#   kernel :
#     func : top_k
#   backward : top_k_grad
P
phlrain 已提交
1155

P
update  
phlrain 已提交
1156 1157 1158 1159 1160 1161 1162 1163 1164
# trace
- api : trace
  args : (Tensor x, int offset, int axis1, int axis2)
  output : Tensor
  infer_meta :
    func : TraceInferMeta
  kernel :
    func : trace
  backward : trace_grad
P
phlrain 已提交
1165

P
phlrain 已提交
1166
# # phi_transfer_layout | not have python api
P
phlrain 已提交
1167

P
phlrain 已提交
1168 1169 1170 1171 1172 1173 1174 1175
# # truncated_gaussian_random
# - api : truncated_gaussian_random
#   args : (int[] shape, float mean, float std, int seed, DataType dtype)
#   output : Tensor
#   infer_meta :
#     func : TruncatedGaussianRandomInferMeta
#   kernel :
#     func : truncated_gaussian_random
P
phlrain 已提交
1176

P
phlrain 已提交
1177 1178 1179 1180 1181 1182 1183 1184
# # unbind
# - api : unbind
#   args : (Tensor x, int axis)
#   output : Tensor[]
#   infer_meta :
#     func : UnbindInferMeta
#   kernel :
#     func : unbind
P
phlrain 已提交
1185

P
update  
phlrain 已提交
1186 1187 1188 1189 1190 1191 1192 1193 1194
# unfold
- api : unfold
  args : (Tensor x, int[] kernel_sizes, int[] strides, int[] paddings, int[] dilations)
  output : Tensor
  infer_meta :
    func : UnfoldInferMeta
  kernel :
    func : unfold
  backward : unfold_grad
P
phlrain 已提交
1195 1196

# # uniform_random_raw selected rows ??
P
update  
phlrain 已提交
1197 1198

# viterbi_decode
P
phlrain 已提交
1199 1200 1201 1202 1203 1204 1205
# - api : viterbi_decode
#   args : (Tensor input, Tensor transition, Tensor length, bool include_bos_eos_tag)
#   output : Tensor(scores), Tensor(path)
#   infer_meta :
#     func : ViterbiDecodeInferMeta
#   kernel :
#     func : viterbi_decode
P
phlrain 已提交
1206

P
update  
phlrain 已提交
1207 1208 1209 1210 1211 1212 1213 1214
# where_index
- api : where_index
  args : (Tensor condition)
  output : Tensor
  infer_meta :
    func : WhereIndexInferMeta
  kernel :
    func : where_index
P
phlrain 已提交
1215 1216


P
phlrain 已提交
1217 1218 1219 1220 1221 1222 1223 1224
# # yolo_box
# - api : yolo_box
#   args : (Tensor x, Tensor img_size, int[] anchors, int class_num, float conf_thresh, int downsample_ratio, bool clip_bbox, float scale_x_y, bool iou_aware, float iou_aware_factor)
#   output : Tensor(boxes), Tensor(scores)
#   infer_meta :
#     func : YoloBoxInferMeta
#   kernel :
#     func : yolo_box