op_compat.yaml 47.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
# All the configuration in this file are only for existing operators,
# which cannot be modified in principle. There's no need to configure
# this file for new operator.
#
# This file is used for two purposes:
# 1. Configure the mapping relationship of parameter names of operator
#    between the operators in ops.yaml and the old operators defined
#    in fluid.
# 2. Save the extra parameters in the OpMaker of operators temporarily,
#    which will be removed in the future.

12 13 14 15 16
# - op : rnn
#   backward : rnn_grad
#   extra :
#     attrs : [bool is_test = false]

17
- op : abs
18
  backward : abs_grad
19 20 21 22
  inputs :
    x : X
  outputs :
    out : Out
23
  extra :
H
HongyuJia 已提交
24
    attrs : [bool use_mkldnn = false]
25

26 27 28 29 30 31
- op : accuracy
  inputs :
    {x : Out , indices : Indices, label: Label}
  outputs :
    {accuracy : Accuracy, correct : Correct, total : Total}

32 33 34 35 36 37
- op : acos
  inputs :
    x : X
  outputs :
    out : Out

38
- op : acosh
39 40 41 42
  inputs :
    x : X
  outputs :
    out : Out
43 44 45 46
  backward : acosh_grad
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

47 48 49 50 51 52
- op : adagrad_
  inputs :
    { param : Param, grad : Grad, moment : Moment, learning_rate : LearningRate, master_param : MasterParam }
  outputs :
    { param_out : ParamOut, moment_out : MomentOut, master_param_out : MasterParamOut }

53 54 55 56 57 58
- op : adamax_
  inputs :
    {param : Param, grad: Grad, learning_rate : LearningRate, moment : Moment, inf_norm : InfNorm, beta1_pow : Beta1Pow, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, moment_out : MomentOut, inf_norm_out : InfNormOut, master_param_out : MasterParamOut}

59 60 61 62 63 64
- op : add (elementwise_add)
  backward : add_grad (elementwise_add_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

65
- op : addmm
66
  backward : addmm_grad
67 68 69 70 71 72
  inputs :
    {input : Input, x : X, y : Y}
  outputs :
    out : Out
  attrs :
    {alpha : Alpha, beta : Beta}
73 74 75
  extra :
    attrs : [bool use_mkldnn = false]

76
- op : affine_grid
77
  backward : affine_grid_grad
78 79 80 81 82 83 84 85
  inputs :
    input : Theta
  outputs :
    out : Output
  int_array:
    output_shape :
      data_type : int
      tensor_name : OutputShape
86 87 88
  extra :
    attrs : [bool use_cudnn = true]

89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
- op : all (reduce_all)
  inputs:
    x : X
  attrs:
    { axis : dim,  keepdim : keep_dim}
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
      support_tensor : true
  manual_signature : [all]
  extra :
    attrs : [bool use_mkldnn = false]

104 105 106 107 108 109 110 111 112 113 114 115 116
- op : allclose
  inputs :
    {x : Input, y : Other}
  outputs :
    out : Out
  scalar :
    rtol :
      data_type : std::string
      tensor_name : Rtol
    atol :
      data_type : std::string
      tensor_name : Atol

117
- op : angle
118
  backward : angle_grad
119 120 121 122
  inputs :
    x : X
  outputs :
    out : Out
123
  extra :
H
HongyuJia 已提交
124
    attrs : [bool use_mkldnn = false]
125

126 127 128 129 130 131
- op : arange(range)
  inputs :
    {start : Start, end : End, step : Step}
  outputs :
    out : Out

engineer1109's avatar
engineer1109 已提交
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
- op : argmax(arg_max)
  inputs :
    x : X
  outputs :
    out : Out
  scalar:
    axis:
      data_type : int64_t
      support_tensor : true

- op : argmin(arg_min)
  inputs :
    x : X
  outputs :
    out : Out
  scalar:
    axis:
      data_type : int64_t
      support_tensor : true

152 153 154 155 156 157 158
- op : argsort
  inputs :
    x : X
  outputs :
    out : Out
    indices : Indices

159 160 161 162 163 164 165 166 167 168 169 170
- op : as_complex
  inputs :
    x : X
  outputs :
    out : Out

- op : as_real
  inputs :
    x : X
  outputs :
    out : Out

171 172 173 174 175 176
- op : asin
  inputs :
    x : X
  outputs :
    out : Out

177
- op : asinh
178
  backward : asinh_grad
179 180 181 182
  inputs :
    x : X
  outputs :
    out : Out
183 184 185
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

186 187 188 189 190 191 192 193 194 195
- op : assign
  backward : assign_grad
  inputs :
    x : X
  outputs :
    out : Out
  manual_signature : [assign, assign_grad]
  get_expected_kernel_type :
    assign : GetAssignExpectedKernelType

196 197 198 199 200
- op : assign_value
  outputs :
    out : Out
  manual_signature : [assign_value]

201 202 203 204 205 206
- op : atan
  inputs :
    x : X
  outputs :
    out : Out

207
- op : atan2
208
  inputs :
209
    {x : X1, y : X2}
210 211 212
  outputs :
    out : Out

213
- op : atanh
214
  backward : atanh_grad
215 216 217 218
  inputs :
    x : X
  outputs :
    out : Out
219 220 221
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

222 223 224 225 226 227
- op : auc
  inputs :
    {x : Predict, label : Label, stat_pos : StatPos, stat_neg : StatNeg, ins_tag_weight : InsTagWeight}
  outputs :
    {auc : AUC, stat_pos_out : StatPosOut, stat_neg_out : StatNegOut}

228
- op : batch_norm
229
  backward : batch_norm_grad
230 231 232 233 234 235
  inputs:
    x : X
    mean : Mean
    variance : Variance
    scale : Scale
    bias : Bias
C
cyber-pioneer 已提交
236 237 238 239 240 241 242
  outputs :
    out : Y
    mean_out: MeanOut
    variance_out: VarianceOut
    saved_mean: SavedMean
    saved_variance: SavedVariance
    reserve_space: ReserveSpace
243 244 245
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

246 247 248 249 250 251 252
- op : bce_loss
  backward : bce_loss_grad
  inputs :
    {input : X, label : Label}
  outputs :
    out : Out

253
- op : bernoulli
254 255 256 257 258
  inputs :
    x : X
  outputs :
    out : Out

259
- op : bicubic_interp (bicubic_interp_v2)
260
  backward : bicubic_interp_grad (bicubic_interp_v2_grad)
261 262 263 264
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
265 266 267
  extra :
    attrs : [bool use_mkldnn = false]

268 269 270 271 272 273
- op : bilinear (bilinear_tensor_product)
  inputs :
    {x : X, y : Y,weight: Weight, bias: Bias}
  outputs :
    {out : Out}

274
- op : bilinear_interp (bilinear_interp_v2)
275
  backward : bilinear_interp_grad (bilinear_interp_v2_grad)
276 277 278 279
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
280 281 282
  extra :
    attrs : [bool use_mkldnn = false]

283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
- op : bitwise_and
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}

- op : bitwise_not
  inputs :
    {x : X}
  outputs :
    {out : Out}

- op : bitwise_or
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}

- op : bitwise_xor
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}

307 308 309 310 311 312
- op : bmm
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

313 314 315 316 317 318 319 320
- op : broadcast_tensors
  backward : broadcast_tensors_grad
  inputs :
    input : X
  outputs :
    out : Out
  drop_empty_grad : [input_grad]

321
- op : ceil
322
  backward : ceil_grad
323 324 325 326
  inputs :
    x : X
  outputs :
    out : Out
327 328 329
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

330 331 332 333 334 335 336
- op : celu
  backward : celu_grad, celu_double_grad(celu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out

337
- op : cholesky
338 339 340 341 342
  inputs :
    x : X
  outputs :
    out : Out

343
- op : cholesky_solve
344 345 346 347 348
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

349
- op : clip
350 351 352 353 354 355 356 357 358 359 360 361
  backward : clip_grad, clip_double_grad
  inputs :
    x : X
  outputs :
    out : Out
  scalar :
    min :
      data_type : float
      tensor_name : Min
    max :
      data_type :  float
      tensor_name : Max
362 363 364
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

365 366 367 368 369 370
- op : clip_by_norm
  inputs :
    x : X
  outputs :
    out : Out

371 372 373 374 375 376 377
- op : complex
  backward : complex_grad
  inputs :
    {real : X, imag : Y}
  outputs :
    out : Out

378
- op : concat
379
  backward : concat_grad
380 381 382 383 384 385 386 387 388 389
  inputs:
    x: X
  outputs:
    out: Out
  attrs:
    axis: axis
  scalar :
    axis :
      data_type : int
      tensor_name : AxisTensor
390 391 392
  extra :
    attrs : [bool use_mkldnn = false, bool use_quantizer = false, str mkldnn_data_type = "float32"]

393 394 395 396 397
- op : conditional_block
  backward : conditional_block_grad
  extra :
    attrs : ['str[] skip_eager_deletion_vars = {}']

398 399 400 401 402 403
- op : conj
  inputs :
    x : X
  outputs :
    out : Out

404
- op : conv2d
405
  backward : conv2d_grad
406
  extra :
407
    attrs : [bool is_test = false, bool use_cudnn = true, bool use_mkldnn = false, bool use_addto = false,
408
             bool force_fp32_output = false,
409
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
410

411
- op : conv2d_fusion
F
Feiyu Chan 已提交
412
  extra :
413
    attrs : [bool is_test = false, bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
F
Feiyu Chan 已提交
414
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
415
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
F
Feiyu Chan 已提交
416 417
             bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
             float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}', bool force_fp32_output = false,
418
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
419

420
- op : conv2d_transpose
421 422 423 424 425
  backward : conv2d_transpose_grad
  extra :
    attrs : [bool is_test = false, bool use_cudnn = true, bool use_mkldnn = false, bool force_fp32_output = false,
             str mkldnn_data_type = "float32", bool fuse_relu = false,
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f,
426
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
427

428
- op : conv3d
429 430 431 432 433
  backward : conv3d_grad
  extra :
    attrs : [bool is_test = false, bool use_cudnn = true, bool use_mkldnn = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f,
             bool use_addto = false, bool fuse_residual_connection = false, bool force_fp32_output = false,
434
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
435

436
- op : conv3d_transpose
437 438
  backward : conv3d_transpose_grad
  extra :
439
    attrs : [bool use_cudnn = true, bool use_mkldnn = false, int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
F
Feiyu Chan 已提交
440

441
- op : cos
442
  backward : cos_grad, cos_double_grad, cos_triple_grad
443 444 445 446
  inputs :
    x : X
  outputs :
    out : Out
447 448 449
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

450
- op : cosh
451
  backward : cosh_grad
452 453 454 455
  inputs :
    x : X
  outputs :
    out : Out
456 457 458
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474
- op : crop (crop_tensor)
  backward : crop_grad (crop_tensor_grad)
  inputs :
    x : X
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int
      tensor_name : Shape
      tensors_name : ShapeTensor
    offsets :
      data_type : int
      tensor_name : Offsets
      tensors_name : OffsetsTensor

475
- op : cross
476 477
  inputs :
    {x : X, y : Y}
478 479 480 481 482
  attrs :
    axis : dim
  outputs :
    out : Out

483 484 485 486 487 488 489
- op : cross_entropy_with_softmax (softmax_with_cross_entropy)
  backward : cross_entropy_with_softmax_grad (softmax_with_cross_entropy_grad)
  inputs :
    {input : Logits, label : Label}
  outputs :
    {softmax : Softmax, loss : Loss}

490 491 492 493 494 495 496 497 498
- op : cumprod
  backward : cumprod_grad
  inputs :
    x : X
  attrs :
    dim : dim
  outputs :
    out : Out

G
GGBond8488 已提交
499 500 501 502 503 504 505 506 507 508 509
- op : cumsum
  backward: cumsum_grad
  inputs :
    x : X
  outputs :
    out : Out
  scalar:
    axis:
      data_type : int
      tensor_name: AxisTensor

510
- op : data_norm
511 512 513 514
  backward : data_norm_grad
  extra :
    attrs : [bool use_mkldnn = false]

515 516 517 518 519 520
- op : decode_jpeg
  inputs :
    x : X
  outputs :
    out : Out

521
- op : depthwise_conv2d
522 523
  backward : depthwise_conv2d_grad
  extra :
524
    attrs : [bool is_test = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
525 526 527 528
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
             bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
             float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}', bool force_fp32_output = false,
529
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
530

531
- op : depthwise_conv2d_transpose
532 533 534 535 536
  backward : depthwise_conv2d_transpose_grad
  extra :
    attrs : [bool is_test = false, bool use_cudnn = false, bool use_mkldnn = false, bool force_fp32_output = false,
             str mkldnn_data_type = "float32", bool fuse_relu = false,
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f,
537
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
538

539 540 541 542
- op : dequantize_linear
  extra :
    attrs : [float moving_rate = 0.9]

543 544 545 546 547 548 549
- op : det (determinant)
  backward : det_grad (determinant_grad)
  inputs :
    x : Input
  outputs :
    out : Out

550
- op : diag (diag_v2)
551
  backward : diag_grad (diag_v2_grad)
552 553 554 555 556
  inputs :
    x : X
  outputs :
    out : Out

557 558 559 560 561 562
- op : diag_embed
  inputs :
    input : Input
  outputs :
    out : Out

563
- op : diagonal
564 565 566 567 568
  inputs :
    x : Input
  outputs :
    out : Out

569
- op : digamma
570 571 572 573 574
  inputs :
    x : X
  outputs :
    out : Out

575 576 577 578 579 580
- op : dirichlet
  inputs :
    alpha : Alpha
  outputs :
    out : Out

581
- op : dist
582 583 584 585 586
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

587 588 589 590
- op : distributed_push_sparse
  extra :
    attrs : ['int[] slots = {}']

591 592
- op : divide (elementwise_div)
  backward : divide_grad (elementwise_div)
593 594 595 596
  inputs :
    {x: X, y : Y}
  outputs :
    out: Out
597 598 599 600
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

601
- op : dot
602 603 604 605 606
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

607
- op : dropout
608
  backward : dropout_grad
C
cyber-pioneer 已提交
609 610 611 612 613 614 615 616 617 618 619
  inputs :
    x : X
  outputs :
    out : Out
    mask : Mask
  attrs :
    p : dropout_prob
    is_test : is_test
    mode : dropout_implementation
    seed : seed
    fix_seed : fix_seed
620 621 622
  extra :
    attrs : [bool fix_seed = false, int seed = 0]

623
- op : dropout_nd
624 625 626 627
  backward : dropout_nd_grad
  extra :
    attrs : [bool fix_seed = false, int seed = 0]

628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647
- op : eig
  inputs :
    x : X
  outputs :
    out_w : Eigenvalues
    out_v : Eigenvectors

- op : eigh
  inputs :
    x : X
  outputs :
    out_w : Eigenvalues
    out_v : Eigenvectors

- op : eigvals
  inputs :
    x : X
  outputs :
    out : Out

648 649 650 651 652 653 654 655 656
- op : eigvalsh
  backward : eigvalsh_grad
  inputs :
    {x : X}
  outputs :
    {eigenvalues : Eigenvalues, eigenvectors : Eigenvectors}
  attrs :
    uplo : UPLO

657 658 659 660 661 662
- op : elementwise_pow
  backward : elementwise_pow_grad
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

663
- op : elu
664 665 666 667 668
  backward : elu_grad, elu_double_grad (elu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
669 670 671
  extra :
    attrs : [bool use_mkldnn = false]

672 673
- op : embedding (lookup_table_v2)
  backward : embedding_grad (lookup_table_v2_grad)
674 675 676 677 678
  inputs :
    {x : Ids, weight : W}
  outputs :
    out : Out
  manual_signature : [embedding_grad]
679 680 681 682 683
  extra :
    attrs : [bool is_sparse = false, bool is_distributed = false, bool remote_prefetch = false,
             int trainer_id = 0, int slot = 0, 'int64_t[] height_sections = {}', 'str[] epmap = {}',
             'str[] table_names = {}']

684 685 686 687 688 689 690 691 692
- op : empty
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList

693 694 695 696 697 698
- op : equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

699 700 701 702 703 704
- op : equal_all
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

705
- op : erf
706 707 708 709 710
  inputs :
    x : X
  outputs :
    out : Out

711
- op : erfinv
712 713 714 715 716
  inputs :
    x : X
  outputs :
    out : Out

717
- op : exp
718
  backward : exp_grad
719 720 721 722
  inputs :
    x : X
  outputs :
    out : Out
723 724
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
725

726 727
- op : expand (expand_v2)
  backward : expand_grad (expand_v2_grad)
728 729 730 731 732 733 734 735 736 737 738
  inputs :
    x : X
  attrs :
   shape : shape
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int
      tensor_name : Shape
      tensors_name : expand_shapes_tensor
739 740 741
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

C
cyber-pioneer 已提交
742
- op : expand_as (expand_as_v2)
743
  backward : expand_as_grad (expand_as_v2_grad)
C
cyber-pioneer 已提交
744
  inputs :
745
    {x : X, y : Y}
C
cyber-pioneer 已提交
746 747 748
  outputs :
    out : Out

749
- op : expm1
750
  backward : expm1_grad
751 752 753 754
  inputs :
    x : X
  outputs :
    out : Out
755 756 757
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

758 759 760 761 762 763 764 765 766
- op : eye
  outputs :
    out : Out
  scalar :
    num_rows :
      support_tensor : true
    num_columns :
      support_tensor : true

767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794
- op : fake_channel_wise_quantize_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_channel_wise_quantize_dequantize_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_quantize_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_quantize_dequantize_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_quantize_dequantize_moving_average_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_quantize_moving_average_abs_max
  extra :
    attrs : [int round_type = 1]

- op : fake_quantize_range_abs_max
  extra :
    attrs : [int round_type = 1]

795
- op : fft_c2c
796 797 798
  inputs: {x: X}
  outputs: {out: Out}

799
- op : fft_c2r
800 801 802
  inputs: {x: X}
  outputs: {out: Out}

803
- op : fft_r2c
804 805 806
  inputs: {x: X}
  outputs: {out: Out}

807 808 809 810 811 812 813
- op : fill_diagonal
  backward : fill_diagonal_grad
  inputs :
    x : X
  outputs :
    out : Out

814 815 816 817 818 819
- op : fill_diagonal_tensor
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

C
cyber-pioneer 已提交
820
- op : flatten (flatten_contiguous_range)
821
  backward : flatten_grad (flatten_contiguous_range_grad)
C
cyber-pioneer 已提交
822 823 824 825 826 827
  inputs :
    x : X
  outputs :
    {out : Out, xshape : XShape}
  attrs :
    {start_axis : start_axis, stop_axis : stop_axis}
828 829 830
  extra :
    outputs : [xshape]
  manual_signature : [flatten, flatten_grad]
C
cyber-pioneer 已提交
831

832 833 834 835 836 837
- op : flip
  inputs :
    x : X
  outputs :
    out : Out

838 839
- op : floor
  backward : floor_grad
840 841 842 843
  inputs :
    x : X
  outputs :
    out : Out
844 845 846
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863
- op : floor_divide (elementwise_floordiv)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

- op : fmax (elementwise_fmax)
  backward : fmax_grad (elementwise_fmax_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

- op : fmin (elementwise_fmin)
  backward : fmin_grad (elementwise_fmin_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

864 865 866 867 868 869
- op : fold
  inputs :
    x : X
  outputs :
    out : Y

870 871 872 873 874 875 876
- op : frame
  backward : frame_grad
  inputs :
    x : X
  outputs :
    out : Out

877
- op : frobenius_norm
878
  backward : frobenius_norm_grad
879 880 881 882 883 884 885 886 887 888 889 890 891
  inputs:
    x : X
  attrs:
    { axis : dim,  keepdim : keep_dim}
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
      support_tensor : true
  get_expected_kernel_type :
    frobenius_norm : GetReduceExpectedKernelType
    frobenius_norm_grad : GetReduceGradExpectedKernelType
892 893 894
  extra :
    attrs : [bool use_mkldnn = false]

895 896 897 898
- op : full (fill_constant)
  extra :
    attrs : [bool use_mkldnn = false]

C
cyber-pioneer 已提交
899 900 901 902 903 904 905 906
- op : full_like (fill_any_like)
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    {value: value, dtype: dtype}

907 908 909 910 911
- op : fused_conv2d
  extra :
    attrs : [bool use_cudnn = false, float fuse_alpha = 0.0f, float fuse_beta = 0.0f, float Scale_in = 1.0f,
             float Scale_out = 1.0f, float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}']

912 913 914 915
- op : fused_transpose
  extra :
    attrs : [str data_format = "AnyLayout"]

916 917 918 919 920
- op : gather
  backward : gather_grad
  extra :
    attrs : [bool overwrite = true]

921 922 923 924 925 926 927
- op : gather_nd
  backward : gather_nd_grad
  inputs :
    {x : X, index : Index}
  outputs :
    out : Out

928 929 930 931 932 933
- op : gather_tree
  inputs :
    {ids : Ids, parents : Parents}
  outputs :
    out : Out

934 935 936 937 938 939 940 941 942 943 944 945
- op : gaussian (gaussian_random)
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList
  extra :
    attrs : [bool use_mkldnn = false]
  manual_signature : [gaussian]

946
- op : gelu
947
  backward : gelu_grad
948 949 950 951
  inputs :
    x : X
  outputs :
    out : Out
952
  extra :
H
HongyuJia 已提交
953
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
954

955 956 957 958 959
- op : grad_add
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

960 961 962 963 964 965 966 967 968 969 970 971
- op : greater_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- op : greater_than
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

972 973 974 975 976 977
- op : grid_sample(grid_sampler)
  backward : grid_sample_grad (grid_sampler_grad)
  inputs :
    {x : X, grid : Grid}
  outputs :
    out : Output
978 979 980
  extra :
    attrs : [bool use_cudnn = true]

C
cyber-pioneer 已提交
981 982 983 984 985 986 987 988 989 990
- op : group_norm
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    y : Y
    mean : Mean
    variance : Variance

991
- op : gru
992 993 994 995
  backward : gru_grad
  extra :
    attrs : [bool is_test = false]

996 997 998 999 1000 1001
- op : gumbel_softmax
  inputs :
    x : X
  outputs :
    out : Out

1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
- op : hardshrink (hard_shrink)
  backward : hardshrink_grad (hard_shrink_grad)
  inputs :
    x : X
  outputs :
    out : Out

- op : hardsigmoid (hard_sigmoid)
  backward : hardsigmoid_grad (hard_sigmoid_grad)
  inputs :
    x : X
  outputs :
    out : Out

C
cyber-pioneer 已提交
1016 1017 1018 1019 1020 1021 1022 1023 1024
- op : hardswish (hard_swish)
  inputs :
    x : X
  outputs :
    out : Out
  backward : hard_swish_grad
  extra :
    attrs : [bool use_mkldnn = false]

1025 1026 1027 1028 1029 1030 1031
- op : hardtanh (brelu)
  backward : hardtanh_grad (brelu_grad)
  inputs :
    x : X
  outputs :
    out : Out

1032 1033 1034 1035 1036 1037
- op : heaviside (elementwise_heaviside)
  backward : heaviside_grad (elementwise_heaviside_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1038 1039 1040 1041 1042 1043
- op : histogram
  inputs :
    input : X
  outputs :
    out : Out

1044 1045 1046 1047 1048 1049 1050
- op : huber_loss
  backward : huber_loss_grad
  inputs :
    {input : X, label : Y}
  outputs :
    {out : Out, residual : Residual}

1051 1052 1053 1054 1055 1056 1057
- op : imag
  backward : imag_grad
  inputs :
    x : X
  outputs :
    out : Out

1058 1059 1060 1061 1062 1063
- op : index_add
  inputs :
    {x : X, index : Index, add_value : AddValue}
  outputs :
    out : Out

1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
- op : index_sample
  inputs :
    {x : X, index : Index}
  outputs :
    out : Out

- op : index_select
  inputs :
    {x : X, index : Index}
  outputs :
    out : Out
  attrs :
    axis : dim

1078
- op : inplace_abn
1079 1080 1081 1082
  backward : inplace_abn_grad
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

C
cyber-pioneer 已提交
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
- op : instance_norm
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    y : Y
    saved_mean : SavedMean
    saved_variance : SavedVariance

1093 1094 1095 1096 1097 1098
- op : inverse
  inputs :
    x : Input
  outputs :
    out : Output

1099 1100 1101 1102 1103 1104
- op : is_empty
  inputs :
    x : X
  outputs :
    out : Out

1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117
- op : isclose
  inputs :
    {x : Input, y : Other}
  outputs :
    out : Out
  scalar :
    rtol :
      data_type : std::string
      tensor_name : Rtol
    atol :
      data_type : std::string
      tensor_name : Atol

1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135
- op : isfinite (isfinite_v2)
  inputs :
    x : X
  outputs :
    out : Out

- op : isinf (isinf_v2)
  inputs :
    x : X
  outputs :
    out : Out

- op : isnan (isnan_v2)
  inputs :
    x : X
  outputs :
    out : Out

1136 1137 1138 1139 1140 1141 1142
- op : kldiv_loss
  backward : kldiv_loss_grad
  inputs :
    {x : X, label : Target}
  outputs :
    out : Loss

1143 1144 1145 1146 1147 1148 1149 1150
- op : kron
  backward : kron_grad
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
  complex_promote : [X, Y]

1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162
- op : kthvalue
  inputs :
    x : X
  outputs :
    {out : Out, indices : Indices}

- op : label_smooth
  inputs :
    {label : X, prior_dist : PriorDist}
  outputs :
    out : Out

1163 1164 1165 1166 1167 1168
- op : lamb_
  inputs :
    {param : Param, grad : Grad, learning_rate : LearningRate, moment1 : Moment1, moment2 : Moment2, beta1_pow : Beta1Pow, beta2_pow : Beta2Pow, master_param : MasterParam, skip_update : SkipUpdate}
  outputs :
    {param_out : ParamOut, moment1_out : Moment1Out, moment2_out : Moment2Out, beta1_pow_out : Beta1PowOut, beta2_pow_out : Beta2PowOut, master_param_outs : MasterParamOut}

1169
- op : layer_norm
1170
  backward : layer_norm_grad
C
cyber-pioneer 已提交
1171 1172 1173 1174 1175 1176 1177 1178
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    out : Y
    mean : Mean
    variance : Variance
1179 1180 1181
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]

1182
- op : leaky_relu
1183 1184 1185 1186 1187 1188 1189
  backward : leaky_relu_grad, leaky_relu_double_grad (leaky_relu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
  attrs:
    negative_slope : alpha
1190 1191 1192
  extra :
    attrs : [bool use_mkldnn = false]

1193 1194 1195 1196 1197 1198 1199
- op : lerp
  backward : lerp_grad
  inputs :
    {x : X, y : Y, weight : Weight}
  outputs :
    out : Out

1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211
- op : less_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- op : less_than
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

1212
- op : lgamma
1213 1214 1215 1216 1217
  inputs :
    x : X
  outputs :
    out : Out

1218
- op : linear_interp (linear_interp_v2)
1219
  backward : linear_interp_grad (linear_interp_v2_grad)
1220 1221 1222 1223
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
1224 1225 1226
  extra :
    attrs : [bool use_mkldnn = false]

1227 1228 1229 1230 1231 1232
- op : linspace
  inputs :
    {start : Start, stop : Stop, number : Num}
  outputs :
    out : Out

1233
- op : log
1234 1235 1236 1237 1238
  backward : log_grad, log_double_grad (log_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
1239 1240 1241
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1242
- op : log10
1243
  backward : log10_grad
1244 1245 1246 1247
  inputs :
    x : X
  outputs :
    out : Out
1248 1249 1250
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1251
- op : log1p
1252
  backward : log1p_grad
1253 1254 1255 1256
  inputs :
    x : X
  outputs :
    out : Out
1257 1258 1259
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1260
- op : log2
1261
  backward : log2_grad
1262 1263 1264 1265
  inputs :
    x : X
  outputs :
    out : Out
1266 1267 1268
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1269 1270 1271 1272 1273 1274 1275
- op : log_loss
  backward : log_loss_grad
  inputs :
    {input : Predicted, label : Labels}
  outputs :
    out : Loss

1276
- op : log_softmax
1277
  backward : log_softmax_grad
1278 1279 1280 1281
  inputs :
    x : X
  outputs :
    out: Out
1282 1283 1284
  extra :
    attrs : [bool use_mkldnn = false]

1285 1286 1287 1288 1289 1290 1291
- op : logcumsumexp
  backward : logcumsumexp_grad
  inputs :
    x : X
  outputs :
    out : Out

1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
- op : logical_and
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- op : logical_not
  inputs :
    x : X
  outputs :
    out : Out

- op : logical_or
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- op : logical_xor
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

1316 1317 1318 1319 1320 1321 1322
- op : logit
  inputs :
    x : X
  outputs :
    out : Out

- op : logsigmoid
1323
  backward : logsigmoid_grad
1324 1325 1326 1327
  inputs :
    x : X
  outputs :
    out : Out
1328 1329 1330
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1331
- op : lrn
1332 1333 1334 1335
  backward : lrn_grad
  extra :
    attrs : [bool use_mkldnn = false, bool is_test = false]

1336 1337 1338 1339 1340 1341 1342
- op : lu_unpack
  backward : lu_unpack_grad
  inputs :
    {x : X, y : Pivots}
  outputs :
    {pmat : Pmat, l : L, u : U}

1343 1344 1345 1346 1347 1348 1349
- op : margin_cross_entropy
  backward : margin_cross_entropy_grad
  inputs:
    {logits : Logits, label : Label}
  outputs:
    {softmax : Softmax, loss : Loss}

1350 1351 1352 1353 1354 1355
- op : masked_select
  inputs :
    {x : X, mask : Mask}
  outputs :
    out : Y

1356
- op : matmul (matmul_v2)
1357
  backward : matmul_grad (matmul_v2_grad)
1358 1359 1360 1361 1362 1363
  inputs :
    {x : X, y : Y}
  attrs :
    {transpose_x : trans_x, transpose_y : trans_y}
  outputs :
    out : Out
1364
  extra :
1365
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
1366

1367 1368 1369 1370 1371 1372
- op : matmul_with_flatten (mul)
  backward : matmul_with_flatten_grad (mul_grad)
  extra :
    attrs : [bool use_mkldnn = false, float scale_x = 1.0f, 'float[] scale_y = {1.0f}',
             float scale_out = 1.0f, bool force_fp32_output = false]

1373 1374 1375 1376 1377 1378 1379 1380
- op : matrix_nms
  inputs :
    {bboxes : BBoxes, scores : Scores}
  outputs :
    {out : Out, index : Index, roisnum : RoisNum}
  get_expected_kernel_type :
    matrix_nms : GetMatrixNmsExpectedKernelType

1381 1382 1383 1384 1385 1386
- op : matrix_power
  inputs :
    x : X
  outputs :
    out : Out

1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400
- op : max (reduce_max)
  backward : max_grad (reduce_max_grad)
  inputs:
    x : X
  attrs:
    { axis : dim,  keepdim : keep_dim}
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
  extra :
    attrs : [bool use_mkldnn = false]

1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412
- op : maximum (elementwise_max)
  backward : maximum_grad (elementwise_max_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

- op : maximum (elementwise_min)
  backward : maximum_grad (elementwise_min_grad)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1413 1414 1415 1416 1417 1418
- op : maxout
  inputs :
    x : X
  outputs :
    out : Out

C
cyber-pioneer 已提交
1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
- op : mean (reduce_mean)
  backward : reduce_mean_grad
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    {axis : dim, keepdim : keep_dim}
  extra :
    attrs : [bool use_mkldnn = false]

1430 1431 1432 1433 1434 1435
- op : merge_selected_rows
  inputs :
    x : X
  outputs :
    out : Out

1436 1437 1438 1439 1440 1441
- op : merged_momentum_
  inputs :
    {param : Param, grad : Grad, velocity : Velocity, learning_rate : LearningRate, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, velocity_out : VelocityOut, master_param_out : MasterParamOut}

1442 1443 1444 1445 1446 1447 1448 1449
- op : meshgrid
  backward : meshgrid_grad
  inputs :
    inputs : X
  outputs :
    out : Out
  drop_empty_grad : [inputs_grad]

1450 1451 1452 1453 1454
- op : mish
  backward : mish_grad
  extra :
    attrs : [bool use_mkldnn = false]

1455 1456 1457 1458 1459 1460 1461
- op : mode
  backward : mode_grad
  inputs :
    x : X
  outputs :
    {out : Out, indices : Indices}

1462 1463 1464 1465 1466 1467
- op : momentum_
  inputs :
    {param : Param, grad : Grad, velocity : Velocity, learning_rate : LearningRate, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, velocity_out : VelocityOut, master_param_out : MasterParamOut}

1468 1469 1470 1471 1472 1473 1474 1475
- op : multi_dot
  backward : multi_dot_grad
  inputs :
    x : X
  outputs :
    out : Out
  drop_empty_grad : [x_grad]

1476 1477 1478 1479 1480 1481 1482 1483 1484 1485
- op : multinomial
  inputs :
    {x : X}
  outputs :
    out : Out
  scalar :
    num_samples :
      data_type : int
      support_tensor : true

1486 1487 1488 1489 1490 1491 1492 1493
- op : multiplex
  backward : multiplex_grad
  inputs :
    {inputs : X, index : Ids}
  outputs :
    out : Out
  drop_empty_grad : [inputs_grad]

1494 1495
- op : multiply (elementwise_mul)
  backward : multiply_grad (elementwise_mul_grad)
1496 1497 1498 1499
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out
1500 1501 1502 1503
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1504
- op : mv
1505 1506 1507 1508 1509
  inputs :
    {x : X, vec : Vec}
  outputs :
    out : Out

Z
zyfncg 已提交
1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521
- op : nanmedian
  backward : nanmedian_grad
  inputs :
    {x : X}
  outputs :
    {out : Out, medians : MedianIndex}
  int_array:
    axis:
      data_type : int
  extra:
    outputs : [medians]

1522 1523 1524 1525 1526 1527
- op : nce
  backward : nce_grad
  extra :
    attrs : [int trainer_id = 0, 'int64_t[] height_sections = {}', 'str[] epmap = {}',
             'str[] table_names = {}', 'int[] custom_neg_classes = {}']

1528
- op : nearest_interp (nearest_interp_v2)
1529
  backward : nearest_interp_grad (nearest_interp_v2_grad)
1530 1531 1532 1533
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
1534 1535 1536
  extra :
    attrs : [bool use_mkldnn = false]

1537 1538 1539 1540 1541 1542 1543
- op : nll_loss
  backward : nll_loss_grad
  inputs :
    {input : X, label : Label, weight : Weight}
  outputs :
    {out : Out, total_weight : Total_weight}

1544 1545 1546 1547 1548 1549 1550 1551
- op : nms
  inputs :
    x : Boxes
  outputs :
    out : KeepBoxesIdxs
  attrs :
    threshold : iou_threshold

C
cyber-pioneer 已提交
1552 1553 1554 1555 1556 1557
- op : nonzero (where_index)
  inputs :
    condition : Condition
  outputs :
    out : Out

1558 1559 1560 1561 1562 1563
- op : not_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

1564 1565 1566 1567 1568 1569
- op : numel(size)
  inputs :
    x : Input
  outputs :
    size : Out

1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
- op : one_hot (one_hot_v2)
  inputs :
    x : X
  outputs :
    out : Out
  scalar :
    depth :
      data_type : int
      tensor_name : depth_tensor

1580 1581 1582 1583 1584 1585 1586
- op : overlap_add
  backward : overlap_add_grad
  inputs :
    x : X
  outputs :
    out : Out

1587 1588 1589 1590 1591 1592 1593
- op : p_norm
  backward: p_norm_grad
  inputs :
    x : X
  outputs :
    out : Out

1594
- op : pad2d
1595 1596 1597 1598
  backward : pad2d_grad
  extra :
    attrs : [bool use_mkldnn = false]

1599
- op : pad3d
1600 1601 1602 1603
  backward : pad3d_grad
  extra :
    attrs : [bool use_mkldnn = false]

1604
- op : partial_sum
1605 1606 1607 1608
  backward : partial_sum_grad
  extra :
    attrs : [bool use_mkldnn = false]

1609 1610 1611 1612 1613 1614 1615
- op : pixel_shuffle
  backward : pixel_shuffle_grad
  inputs :
    x : X
  outputs :
    out : Out

1616
- op : poisson
1617 1618 1619 1620 1621
  inputs :
    x : X
  outputs :
    out : Out

1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
- op : pool2d
  backward : pool2d_grad
  extra :
    attrs : [bool use_mkldnn = false, bool use_quantizer = false,
              str mkldnn_data_type = "float32", bool is_test = false]

- op : pool3d
  backward : pool3d_grad
  extra :
    attrs : [bool use_mkldnn = false]

1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645
- op : pow
  backward : pow_grad, pow_double_grad, pow_triple_grad
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    y : factor
  scalar :
    y :
      data_type : float
      tensor_name : FactorTensor

1646
- op : prelu
1647
  backward : prelu_grad
1648 1649 1650 1651
  inputs :
    { x : X, alpha : Alpha}
  outputs :
    out : Out
1652 1653 1654
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]

1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668
- op : prod (reduce_prod)
  backward : prod_grad (reduce_prod_grad)
  inputs:
    x : X
  attrs:
    { dims : dim,  keep_dim : keep_dim}
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
  extra :
    attrs : [bool use_mkldnn = false]

1669 1670 1671 1672 1673 1674 1675 1676 1677
- op : put_along_axis
  backward : put_along_axis_grad
  inputs :
    {arr : Input, indices : Index, values : Value}
  outputs :
    out : Result
  attrs :
    {axis : Axis, reduce : Reduce}

1678 1679 1680 1681 1682 1683 1684
- op : qr
  backward : qr_grad
  inputs :
    x : X
  outputs :
    {q : Q, r : R}

1685 1686 1687 1688
- op : quantize_linear
  extra :
    attrs : [float moving_rate = 0.9]

1689 1690 1691 1692 1693 1694 1695 1696 1697 1698
- op : randint
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList
  manual_signature : [randint]

1699 1700 1701 1702 1703 1704
- op : randperm
  outputs :
    out : Out
  extra :
    attrs : [int seed = 0]

1705 1706 1707 1708 1709 1710 1711
- op : real
  backward : real_grad
  inputs :
    x : X
  outputs :
    out : Out

1712
- op : reciprocal
1713
  backward : reciprocal_grad
1714 1715 1716 1717
  inputs :
    x : X
  outputs :
    out : Out
1718 1719 1720
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1721
- op : reduce_amax
1722 1723 1724 1725
  backward : reduce_amax_grad
  extra :
    attrs : [bool use_mkldnn = false]

1726
- op : reduce_amin
1727 1728 1729 1730
  backward : reduce_amin_grad
  extra :
    attrs : [bool use_mkldnn = false]

1731
- op : reduce_any
1732 1733 1734
  extra :
    attrs : [bool use_mkldnn = false]

1735
- op : reduce_min
1736 1737 1738 1739
  backward : reduce_min_grad
  extra :
    attrs : [bool use_mkldnn = false]

1740
- op : relu
1741 1742 1743 1744 1745
  backward : relu_grad, relu_double_grad (relu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
1746 1747 1748
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1749
- op : relu6
1750 1751 1752 1753
  backward : relu6_grad
  extra :
    attrs : [bool use_mkldnn = false]

1754 1755 1756 1757 1758
- op : remainder (elementwise_mod)
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1759
- op : renorm
1760
  backward : renorm_grad
1761 1762 1763 1764
  inputs :
    x : X
  outputs :
    out : Out
1765 1766 1767
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
- op : reshape (reshape2)
  backward : reshape_grad (reshape2_grad)
  inputs:
    x : X
  outputs:
    out : Out
  int_array:
    shape :
      data_type : int
      tensor_name : Shape
      tensors_name : ShapeTensor
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool use_quantizer = false]

1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792
- op : reverse
  inputs:
    x : X
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
      support_tensor : true
  manual_signature : [reverse]

1793 1794 1795 1796 1797 1798
- op : rmsprop_
  inputs :
    {param: Param, mean_square: MeanSquare, mean_grad: MeanGrad, learning_rate: LearningRate, grad: Grad, moment: Moment, master_param: MasterParam}
  outputs :
    {param_out: ParamOut, moment_out: MomentOut, mean_square_out: MeanSquareOut, mean_grad_out: MeanGradOut, master_param_outs: MasterParamOut}

1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809
- op : roll
  backward : roll_grad
  inputs :
    x : X
  outputs :
    out : Out
  int_array :
    shifts :
      data_type : int64_t
      tensor_name : ShiftsTensor

1810
- op : round
1811
  backward : round_grad
1812 1813 1814 1815
  inputs :
    x : X
  outputs :
    out : Out
1816
  extra :
1817 1818
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1819
- op : rsqrt
1820 1821 1822 1823 1824
  backward : rsqrt_grad, rsqrt_double_grad (rsqrt_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
1825 1826
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
1827

1828
- op : scale
HappyHeavyRain's avatar
HappyHeavyRain 已提交
1829
  backward : scale_grad
1830 1831 1832
  inputs :
    x : X
  outputs :
HappyHeavyRain's avatar
HappyHeavyRain 已提交
1833 1834 1835 1836 1837
    out : Out
  scalar :
    scale :
      data_type : float
      tensor_name : ScaleTensor
1838 1839 1840
  extra :
    attrs : [bool use_mkldnn = false]

1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
- op : scatter
  backward : scatter_grad
  inputs :
    {x : X, index : Ids, updates : Updates}
  outputs :
    out : Out

- op : scatter_nd_add
  backward : scatter_nd_add_grad
  inputs :
    {x : X, index : Index, updates : Updates}
  outputs :
    out : Out

1855 1856 1857 1858 1859 1860
- op : searchsorted
  inputs :
    {sorted_sequence : SortedSequence, values : Values}
  outputs :
    out : Out

1861
- op : seed
1862 1863 1864
  extra :
    attrs : [bool deterministic = false, str rng_name = "", bool force_cpu = false]

L
lzydev 已提交
1865 1866 1867 1868 1869 1870 1871
- op : segment_pool
  backward : segment_pool_grad
  inputs :
    {x : X, segment_ids : SegmentIds}
  outputs :
    {out : Out, summed_ids : SummedIds}

1872 1873 1874 1875 1876 1877 1878
- op : selu
  backward : selu_grad
  inputs :
    x : X
  outputs :
    out : Out

1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900
- op : send_u_recv(graph_send_recv)
  backward : send_u_recv_grad(graph_send_recv_grad)
  inputs :
    {x : X, src_index : Src_index, dst_index : Dst_index}
  outputs :
    {out : Out, dst_count : Dst_count}
  int_array :
    out_size:
      data_type : int64_t
      tensor_name : Out_size

- op : send_ue_recv(graph_send_ue_recv)
  backward : send_ue_recv_grad(graph_send_ue_recv_grad)
  inputs :
    {x : X, y : Y, src_index : Src_index, dst_index : Dst_index}
  outputs :
    {out : Out, dst_count : Dst_count}
  int_array :
    out_size:
      data_type : int64_t
      tensor_name : Out_size

1901 1902 1903
- op : send_uv (graph_send_uv)
  backward : send_uv_grad (graph_send_uv_grad)

1904 1905 1906 1907 1908
- op : sequence_softmax
  backward : sequence_softmax_grad
  extra :
    attrs : [str data_format = "AnyLayout"]

1909 1910 1911 1912 1913 1914
- op : sgd_
  inputs :
    {param : Param, learning_rate : LearningRate, grad : Grad, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, master_param_out : MasterParamOut}
  get_expected_kernel_type :
1915
    sgd_ : GetSgdExpectedKernelType
1916 1917 1918
  extra :
    attrs : [bool use_mkldnn=false]

1919
- op : shape
1920 1921 1922
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

1923 1924 1925 1926 1927 1928
- op : shard_index
  inputs :
    input : X
  outputs :
    out : Out

1929 1930 1931 1932 1933 1934 1935
- op : share_buffer
  inputs :
    x : X
  outputs :
    out : Out
    xout : XOut

1936
- op : shuffle_channel
1937 1938 1939 1940
  backward : shuffle_channel_grad
  extra :
    attrs : [bool use_mkldnn = false]

1941
- op : sigmoid
1942 1943 1944 1945 1946
  backward : sigmoid_grad, sigmoid_double_grad (sigmoid_grad_grad), sigmoid_triple_grad
  inputs :
    x : X
  outputs :
    out : Out
1947 1948 1949
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

HappyHeavyRain's avatar
HappyHeavyRain 已提交
1950 1951 1952 1953 1954 1955 1956
- op : sign
  backward : sign_grad
  inputs :
    x : X
  outputs :
    out : Out

1957
- op : silu
1958
  backward : silu_grad
1959 1960 1961 1962
  inputs :
    x : X
  outputs :
    out : Out
1963 1964 1965
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1966
- op : sin
1967
  backward : sin_grad, sin_double_grad, sin_triple_grad
1968 1969 1970 1971
  inputs :
    x : X
  outputs :
    out : Out
1972 1973 1974
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1975
- op : sinh
1976
  backward : sinh_grad
1977 1978 1979 1980
  inputs :
    x : X
  outputs :
    out : Out
1981 1982 1983
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1984
- op : slice
1985 1986 1987 1988
  backward : slice_grad
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

1989 1990 1991 1992 1993 1994 1995
- op : slogdet(slogdeterminant)
  backward : slogdet_grad(slogdeterminant_grad)
  inputs :
    x : Input
  outputs :
    out : Out

1996
- op : softmax
1997
  backward : softmax_grad
1998 1999
  inputs :
    x : X
2000
  extra :
2001
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]
2002

2003
- op : softplus
W
will-jl944 已提交
2004
  backward : softplus_grad, softplus_double_grad
2005 2006 2007 2008
  inputs :
    x : X
  outputs :
    out : Out
2009
  extra :
S
Sławomir Siwek 已提交
2010
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
2011

2012 2013 2014 2015 2016 2017 2018 2019 2020
- op : softshrink
  backward : softshrink_grad
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    threshold : lambda

2021
- op : softsign
2022
  backward : softsign_grad
2023 2024 2025 2026
  inputs :
    x : X
  outputs :
    out : Out
2027 2028
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
2029

2030
- op : solve
2031 2032 2033 2034 2035
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

2036 2037 2038 2039 2040 2041 2042
- op : spectral_norm
  backward : spectral_norm_grad
  inputs :
    {weight : Weight, u : U, v : V}
  outputs :
    out : Out

2043
- op : sqrt
2044 2045 2046 2047 2048
  backward : sqrt_grad, sqrt_double_grad (sqrt_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2049 2050 2051
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2052
- op : square
2053 2054 2055 2056 2057
  backward : square_grad, square_double_grad (square_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2058 2059 2060
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2061
- op : squeeze (squeeze2)
2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072
  backward : squeeze_grad (squeeze2_grad), squeeze_double_grad(squeeze2_double_grad)
  inputs :
    x : X
  attrs :
   axis : axes
  outputs :
    {out : Out, xshape : XShape}
  int_array:
    axis :
      data_type : int
      support_tensor : true
2073 2074
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
2075
    outputs : [xshape]
2076

2077
- op : stack
2078
  backward : stack_grad
2079 2080 2081 2082
  inputs :
    x : X
  outputs :
    out : Y
2083 2084
  extra :
    attrs : [bool use_mkldnn = false]
2085
  drop_empty_grad : [x_grad]
2086

Z
zyfncg 已提交
2087 2088 2089 2090 2091 2092 2093
- op : stanh
  backward : stanh_grad
  inputs :
    x : X
  outputs :
    out : Out

2094 2095
- op : subtract (elementwise_sub)
  backward : subtract_grad (elementwise_sub_grad)
2096 2097 2098 2099
  inputs :
    {x : X, y: Y}
  outputs :
    out : Out
2100 2101 2102 2103
  extra :
    attrs : [bool use_mkldnn = false, str x_data_format = "", str y_data_format = "", str mkldnn_data_type = "float32",
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117
- op : sum (reduce_sum)
  backward : (sum_grad) reduce_sum_grad
  inputs:
    {x : X}
  attrs:
    { axis : dim,  keepdim : keep_dim, dtype : out_dtype}
  outputs:
    out : Out
  int_array:
      axis :
        data_type : int
  extra :
    attrs : [bool use_mkldnn = false]

2118 2119 2120 2121 2122 2123 2124
- op : svd
  backward : svd_grad
  inputs :
    x : X
  outputs :
    {u : U, s : S, vh : VH}

2125
- op : swish
2126 2127 2128 2129
  backward : swish_grad
  extra :
    attrs : [bool use_mkldnn = false]

2130
- op : sync_batch_norm
2131 2132 2133 2134
  backward : sync_batch_norm_grad
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

2135 2136 2137 2138 2139 2140 2141 2142 2143
- op : take_along_axis
  backward : take_along_axis_grad
  inputs :
    {arr : Input, indices : Index}
  outputs :
    out : Result
  attrs :
    axis : Axis

2144
- op : tan
2145
  backward : tan_grad
2146 2147 2148 2149
  inputs :
    x : X
  outputs :
    out : Out
2150 2151 2152
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2153
- op : tanh
2154 2155 2156 2157 2158
  backward : tanh_grad, tanh_double_grad (tanh_grad_grad), tanh_triple_grad
  inputs :
    x : X
  outputs :
    out : Out
2159 2160 2161
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2162
- op : tanh_shrink
2163
  backward : tanh_shrink_grad
2164 2165 2166 2167
  inputs :
    x : X
  outputs :
    out : Out
2168 2169 2170
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2171 2172 2173 2174 2175 2176
- op : thresholded_relu
  inputs :
    x : X
  outputs :
    out : Out

2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188
- op : tile
  backward : tile_grad, tile_double_grad
  inputs :
    x : X
  outputs :
    out : Out
  int_array:
    repeat_times :
      data_type : int
      tensor_name : RepeatTimes
      tensors_name : repeat_times_tensor

2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199
- op : topk (top_k_v2)
  backward : topk_grad (top_k_v2_grad)
  inputs :
    x : X
  outputs :
    {out : Out, indices : Indices}
  scalar :
    k :
      data_type : int
      tensor_name : K

2200
- op : trace
2201 2202 2203 2204
  inputs :
    x : Input
  outputs :
    out : Out
2205

2206 2207
- op : transpose (transpose2)
  backward : transpose_grad (transpose2_grad)
J
Jiabin Yang 已提交
2208 2209
  attrs:
    perm : axis
2210
  extra :
J
Jiabin Yang 已提交
2211
    outputs : [XShape]
2212
    attrs : [bool use_mkldnn = false, str data_format = "AnyLayout", str mkldnn_data_type = "float32"]
2213

2214
- op : trilinear_interp (trilinear_interp_v2)
2215
  backward : trilinear_interp_grad (trilinear_interp_v2_grad)
2216 2217 2218 2219
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
2220 2221 2222
  extra :
    attrs : [bool use_mkldnn = false]

2223
- op : trunc
2224
  inputs :
2225
    input : X
2226 2227
  outputs :
    out : Out
2228

2229 2230 2231 2232
- op : truncated_gaussian_random
  outputs :
    out : Out

2233 2234 2235 2236 2237 2238
- op : unbind
  inputs :
    input : X
  outputs :
    out : Out

2239 2240 2241 2242 2243 2244
- op : unfold
  inputs :
    x : X
  outputs :
    out : Y

2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261
- op : uniform (uniform_random)
  outputs :
    out : Out
  int_array :
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList
  scalar :
      min :
        data_type : float
        support_tensor : true
      max :
        data_type : float
        support_tensor : true
  manual_signature : [uniform]

L
lzydev 已提交
2262 2263 2264 2265 2266 2267 2268 2269 2270
- op : unique
  inputs :
    {x : X}
  outputs :
    {out : Out, indices : Indices, inverse : Index, counts : Counts}
  get_expected_kernel_type :
    unique : GetUniqueExpectedKernelType
  manual_signature : [unique]

2271 2272 2273 2274 2275 2276
- op : unique_consecutive
  inputs :
    x : X
  outputs :
    {out : Out, index : Index, counts : Counts}

2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292
- op : unsqueeze (unsqueeze2)
  backward : unsqueeze_grad (unsqueeze2_grad), unsqueeze_double_grad(unsqueeze2_double_grad)
  inputs :
    x : X
  attrs :
   axis : axes
  outputs :
    {out : Out, xshape : XShape}
  int_array:
    axis :
      data_type : int
      tensor_name : AxesTensor
      tensors_name : AxesTensorList
  extra :
    outputs : [xshape]

2293 2294 2295 2296 2297 2298 2299
- op : unstack
  backward : unstack_grad
  inputs :
    x : X
  outputs :
    out : Y

2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
- op : update_loss_scaling_
  inputs :
    {x : X, found_infinite : FoundInfinite, prev_loss_scaling : PrevLossScaling, in_good_steps : InGoodSteps, in_bad_steps : InBadSteps}
  outputs :
    {out : Out, loss_scaling : LossScaling, out_good_steps : OutGoodSteps, out_bad_steps : OutBadSteps}
  scalar :
    stop_update :
      data_type : bool
      tensor_name : StopUpdate
  get_expected_kernel_type :
    update_loss_scaling_ : GetUpdateLossScalingExpectedKernelType

2312 2313 2314 2315 2316 2317
- op : viterbi_decode
  inputs :
    {potentials : Input, transition_params : Transition, lengths : Length}
  outputs :
    {scores : Scores, path : Path}

2318 2319 2320 2321 2322 2323 2324
- op : warpctc
  backward : warpctc_grad
  inputs :
    {logits : Logits, label : Label, logits_length : LogitsLength, labels_length : LabelLength}
  outputs :
    {warpctcgrad : WarpCTCGrad, loss : Loss}

2325 2326 2327 2328 2329 2330 2331
- op : where
  backward : where_grad
  inputs :
    {condition : Condition, x : X, y : Y}
  outputs :
    out : Out

2332 2333
- op : while
  backward : while_grad
2334
  extra :
2335
    attrs : ['str[] skip_eager_deletion_vars = {}']
2336

2337 2338 2339 2340 2341 2342
- op : yolo_box
  inputs :
    {x : X, img_size : ImgSize}
  outputs :
    {boxes : Boxes, scores : Scores}

2343 2344 2345 2346 2347 2348
- op: sigmoid_cross_entropy_with_logits
  backward: sigmoid_cross_entropy_with_logits_grad
  inputs :
    {x: X, label: Label}
  outputs :
    out : Out
2349 2350 2351 2352 2353 2354 2355

- op: squared_l2_norm
  backward: squared_l2_norm_grad
  inputs :
    x : X
  outputs :
    out : Out
2356 2357 2358 2359 2360 2361 2362

- op: temporal_shift
  backward: temporal_shift_grad
  inputs :
    x : X
  outputs :
    out : Out