|
24 | 24 | func : add |
25 | 25 | inplace : (x -> out) |
26 | 26 | backward : add_grad |
| 27 | + data_transform : |
| 28 | + support_trans_dtype : x, y |
27 | 29 | interfaces : paddle::dialect::InferSymbolicShapeInterface |
28 | 30 |
|
29 | 31 | - op : add_n |
|
383 | 385 | kernel : |
384 | 386 | func : divide |
385 | 387 | inplace: (x -> out) |
| 388 | + data_transform : |
| 389 | + support_trans_dtype : x, y |
386 | 390 | backward : divide_grad |
387 | 391 |
|
388 | 392 | - op : dropout |
|
416 | 420 | spmd_rule: ElementwiseBinaryInferSpmd |
417 | 421 | kernel : |
418 | 422 | func : elementwise_pow |
| 423 | + data_transform : |
| 424 | + support_trans_dtype : x, y |
419 | 425 | backward : elementwise_pow_grad |
420 | 426 |
|
421 | 427 | - op : embedding |
|
474 | 480 | spmd_rule: ElementwiseBinaryInferSpmd |
475 | 481 | kernel : |
476 | 482 | func : equal |
| 483 | + data_transform : |
| 484 | + support_trans_dtype : x, y |
477 | 485 | inplace: (x -> out) |
478 | 486 |
|
479 | 487 | - op : exponential_ |
|
521 | 529 | func : ElementwiseInferMeta |
522 | 530 | kernel : |
523 | 531 | func : floor_divide |
| 532 | + data_transform : |
| 533 | + support_trans_dtype : x, y |
524 | 534 | inplace: (x -> out) |
525 | 535 |
|
526 | 536 | - op : frobenius_norm |
|
664 | 674 | func : CompareInferMeta |
665 | 675 | kernel : |
666 | 676 | func : greater_equal |
| 677 | + data_transform : |
| 678 | + support_trans_dtype : x, y |
667 | 679 | inplace: (x -> out) |
668 | 680 |
|
669 | 681 | - op : greater_than |
|
673 | 685 | func : CompareInferMeta |
674 | 686 | kernel : |
675 | 687 | func : greater_than |
| 688 | + data_transform : |
| 689 | + support_trans_dtype : x, y |
676 | 690 | inplace: (x -> out) |
677 | 691 |
|
678 | 692 | - op : hardswish |
|
712 | 726 | func : CompareInferMeta |
713 | 727 | kernel : |
714 | 728 | func : less_equal |
| 729 | + data_transform : |
| 730 | + support_trans_dtype : x, y |
715 | 731 | inplace: (x -> out) |
716 | 732 |
|
717 | 733 | - op : less_than |
|
721 | 737 | func : CompareInferMeta |
722 | 738 | kernel : |
723 | 739 | func : less_than |
| 740 | + data_transform : |
| 741 | + support_trans_dtype : x, y |
724 | 742 | inplace: (x -> out) |
725 | 743 |
|
726 | 744 | - op : linspace |
|
786 | 804 | spmd_rule : MatmulInferSpmd |
787 | 805 | kernel : |
788 | 806 | func : matmul |
| 807 | + data_transform : |
| 808 | + support_trans_dtype : x, y |
789 | 809 | backward : matmul_grad |
790 | 810 |
|
791 | 811 | - op : matrix_rank |
|
823 | 843 | spmd_rule : ElementwiseBinaryInferSpmd |
824 | 844 | kernel : |
825 | 845 | func : maximum |
| 846 | + data_transform : |
| 847 | + support_trans_dtype : x, y |
826 | 848 | backward : maximum_grad |
827 | 849 |
|
828 | 850 | - op : mean |
|
879 | 901 | func : ElementwiseInferMeta |
880 | 902 | kernel : |
881 | 903 | func : minimum |
| 904 | + data_transform : |
| 905 | + support_trans_dtype : x, y |
882 | 906 | backward : minimum_grad |
883 | 907 |
|
884 | 908 | - op : mish |
|
922 | 946 | spmd_rule : ElementwiseBinaryInferSpmd |
923 | 947 | kernel : |
924 | 948 | func : not_equal |
| 949 | + data_transform : |
| 950 | + support_trans_dtype : x, y |
925 | 951 | inplace: (x -> out) |
926 | 952 |
|
927 | 953 | - op : one_hot |
|
1057 | 1083 | func : ElementwiseInferMeta |
1058 | 1084 | kernel : |
1059 | 1085 | func : remainder |
| 1086 | + data_transform : |
| 1087 | + support_trans_dtype : x, y |
1060 | 1088 | inplace : (x -> out) |
1061 | 1089 |
|
1062 | 1090 | - op : repeat_interleave |
|
1274 | 1302 | kernel : |
1275 | 1303 | func : subtract |
1276 | 1304 | inplace : (x -> out) |
| 1305 | + data_transform : |
| 1306 | + support_trans_dtype : x, y |
1277 | 1307 | backward : subtract_grad |
1278 | 1308 | interfaces : paddle::dialect::InferSymbolicShapeInterface |
1279 | 1309 |
|
|
0 commit comments