op_compat.yaml 63.3 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]

H
hong 已提交
47 48 49 50 51 52 53
- op : adadelta_ (adadelta)
  inputs :
    {param : Param, grad: Grad, avg_squared_grad : AvgSquaredGrad, avg_squared_update : AvgSquaredUpdate, learning_rate : LearningRate, master_param : MasterParam }
  outputs :
    {param_out : ParamOut, moment_out : AvgSquaredGradOut, inf_norm_out : AvgSquaredUpdateOut, master_param_out : MasterParamOut}

- op : adagrad_ (adagrad)
54 55 56 57 58
  inputs :
    { param : Param, grad : Grad, moment : Moment, learning_rate : LearningRate, master_param : MasterParam }
  outputs :
    { param_out : ParamOut, moment_out : MomentOut, master_param_out : MasterParamOut }

H
hong 已提交
59
- op : adam_ (adam)
60 61 62 63 64 65 66 67 68 69 70
  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_out: MasterParamOut}
  scalar :
    beta1 :
      data_type : float
      tensor_name : Beta1Tensor
    beta2 :
      data_type : float
      tensor_name : Beta2Tensor
71
    epsilon :
72
      data_type : float
73
      tensor_name : EpsilonTensor
74 75
  manual_signature : [adam_]

H
hong 已提交
76
- op : adamax_ (adamax)
77 78 79 80 81
  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}

H
hong 已提交
82
- op : adamw_ (adamw)
83 84 85 86 87 88 89 90 91 92 93
  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_out: MasterParamOut}
  scalar :
    beta1 :
      data_type : float
      tensor_name : Beta1Tensor
    beta2 :
      data_type : float
      tensor_name : Beta2Tensor
94
    epsilon :
95
      data_type : float
96
      tensor_name : EpsilonTensor
97

98
- op : add (elementwise_add)
99 100 101 102 103
  backward : add_grad (elementwise_add_grad), add_double_grad (elementwise_add_grad_grad), add_triple_grad (elementwise_add_triple_grad)
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
104
  extra :
105
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
106
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
107
  complex_promote : [X, Y]
108

X
xiaoguoguo626807 已提交
109 110 111 112 113 114
- op : add_n (sum)
  inputs:
    {inputs : X}
  outputs:
    {out : Out}

115
- op : addmm
116
  backward : addmm_grad
117 118 119 120 121 122
  inputs :
    {input : Input, x : X, y : Y}
  outputs :
    out : Out
  attrs :
    {alpha : Alpha, beta : Beta}
123 124 125
  extra :
    attrs : [bool use_mkldnn = false]

126
- op : affine_grid
127
  backward : affine_grid_grad
128 129 130
  inputs :
    input : Theta
  outputs :
131
    output : Output
132 133 134 135
  int_array:
    output_shape :
      data_type : int
      tensor_name : OutputShape
136 137 138
  extra :
    attrs : [bool use_cudnn = true]

139 140 141 142 143 144 145 146 147 148 149
- op : all (reduce_all)
  inputs:
    x : X
  attrs:
    { axis : dim,  keepdim : keep_dim}
  outputs:
    out : Out
  manual_signature : [all]
  extra :
    attrs : [bool use_mkldnn = false]

150 151 152 153 154 155 156 157 158 159 160 161 162
- 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

163 164 165 166 167 168 169 170 171 172 173 174 175 176
- op : amax (reduce_amax)
  backward : amax_grad (reduce_amax_grad)
  inputs :
    x : X
  outputs :
    out : Out
  attrs:
    { axis : dim,  keepdim : keep_dim }
  extra :
    attrs : [bool use_mkldnn = false]
  get_expected_kernel_type :
    amax_grad : GetReduceGradExpectedKernelType
  manual_signature : [amax]

177 178 179 180 181 182 183 184 185 186 187 188 189 190
- op : amin (reduce_amin)
  backward : amin_grad (reduce_amin_grad)
  inputs :
    x : X
  outputs :
    out : Out
  attrs:
    { axis : dim,  keepdim : keep_dim }
  extra :
    attrs : [bool use_mkldnn = false]
  get_expected_kernel_type :
    amin_grad : GetReduceGradExpectedKernelType
  manual_signature : [amin]

191
- op : angle
192
  backward : angle_grad
193 194 195 196
  inputs :
    x : X
  outputs :
    out : Out
197
  extra :
H
HongyuJia 已提交
198
    attrs : [bool use_mkldnn = false]
199

200 201 202 203 204 205 206 207 208 209 210 211 212
- op : any (reduce_any)
  inputs :
    x : X
  outputs :
    out : Out
  attrs:
    { axis : dim,  keepdim : keep_dim }
  extra :
    attrs : [bool use_mkldnn = false]
  get_expected_kernel_type :
    any : GetReduceOpUseInputPlaceExpectedKernelType
  manual_signature : [any]

213 214 215 216 217 218
- op : arange(range)
  inputs :
    {start : Start, end : End, step : Step}
  outputs :
    out : Out

engineer1109's avatar
engineer1109 已提交
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238
- 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

239 240 241 242 243 244 245
- op : argsort
  inputs :
    x : X
  outputs :
    out : Out
    indices : Indices

246 247 248 249 250 251 252 253 254 255 256 257
- op : as_complex
  inputs :
    x : X
  outputs :
    out : Out

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

258 259 260 261 262 263
- op : asin
  inputs :
    x : X
  outputs :
    out : Out

264
- op : asinh
265
  backward : asinh_grad
266 267 268 269
  inputs :
    x : X
  outputs :
    out : Out
270 271 272
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

273 274 275 276
- op : assert
  inputs :
    {cond : Cond, data : Data}

277 278 279 280 281 282 283 284 285 286
- op : assign
  backward : assign_grad
  inputs :
    x : X
  outputs :
    out : Out
  manual_signature : [assign, assign_grad]
  get_expected_kernel_type :
    assign : GetAssignExpectedKernelType

287 288 289 290 291
- op : assign_value
  outputs :
    out : Out
  manual_signature : [assign_value]

292 293 294 295 296 297
- op : atan
  inputs :
    x : X
  outputs :
    out : Out

298
- op : atan2
H
hong 已提交
299
  backward : atan2_grad
300
  inputs :
301
    {x : X1, y : X2}
302 303 304
  outputs :
    out : Out

305
- op : atanh
306
  backward : atanh_grad
307 308 309 310
  inputs :
    x : X
  outputs :
    out : Out
311 312 313
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

314 315 316 317 318 319
- 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}

320
- op : batch_norm
321
  backward : batch_norm_grad
322 323 324 325 326 327
  inputs:
    x : X
    mean : Mean
    variance : Variance
    scale : Scale
    bias : Bias
C
cyber-pioneer 已提交
328 329 330 331 332 333 334
  outputs :
    out : Y
    mean_out: MeanOut
    variance_out: VarianceOut
    saved_mean: SavedMean
    saved_variance: SavedVariance
    reserve_space: ReserveSpace
335 336 337
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

338 339 340 341 342 343 344
- op : bce_loss
  backward : bce_loss_grad
  inputs :
    {input : X, label : Label}
  outputs :
    out : Out

345
- op : bernoulli
346 347 348 349 350
  inputs :
    x : X
  outputs :
    out : Out

351
- op : bicubic_interp (bicubic_interp_v2)
352
  backward : bicubic_interp_grad (bicubic_interp_v2_grad)
353 354 355 356
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
357 358 359
  extra :
    attrs : [bool use_mkldnn = false]

360
- op : bilinear (bilinear_tensor_product)
361
  backward: bilinear_grad (bilinear_tensor_product_grad)
362 363 364 365 366
  inputs :
    {x : X, y : Y,weight: Weight, bias: Bias}
  outputs :
    {out : Out}

367
- op : bilinear_interp (bilinear_interp_v2)
368
  backward : bilinear_interp_grad (bilinear_interp_v2_grad)
369 370 371 372
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
373 374 375
  extra :
    attrs : [bool use_mkldnn = false]

376 377 378 379 380 381 382 383 384 385 386 387
- op : bincount
  inputs :
    {x : X, weights : Weights}
  outputs :
    out : Out
  scalar:
    minlength:
      data_type : int
      support_tensor : true
  get_expected_kernel_type :
    bincount : GetBincountExpectedKernelType

388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
- 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}

412 413 414 415 416 417
- op : bmm
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

418 419 420 421 422 423
- op : box_coder
  inputs :
    {prior_box : PriorBox , prior_box_var : PriorBoxVar, target_box: TargetBox}
  outputs :
    output_box : OutputBox

424 425 426 427 428 429 430 431
- op : broadcast_tensors
  backward : broadcast_tensors_grad
  inputs :
    input : X
  outputs :
    out : Out
  drop_empty_grad : [input_grad]

H
hong 已提交
432 433 434 435 436 437
- op : c_concat
  inputs :
    x : X
  outputs :
    out : Out

438 439 440 441 442 443
- op : cast
  inputs :
    x : X
  outputs :
    out : Out

444
- op : ceil
445
  backward : ceil_grad
446 447 448 449
  inputs :
    x : X
  outputs :
    out : Out
450 451 452
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

453 454 455 456 457 458 459
- op : celu
  backward : celu_grad, celu_double_grad(celu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out

460 461 462 463 464 465 466 467
- op : check_finite_and_unscale_
  inputs :
    {x : X, scale: Scale}
  outputs :
    {out : Out, found_infinite: FoundInfinite}
  get_expected_kernel_type :
    check_finite_and_unscale_ : GetCheckFiniteAndUnscaleExpectedKernelType

468
- op : cholesky
469 470 471 472 473
  inputs :
    x : X
  outputs :
    out : Out

474
- op : cholesky_solve
475 476 477 478 479
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

480 481 482 483 484 485
- op : class_center_sample
  inputs :
    label : Label
  outputs :
    {remapped_label : RemappedLabel, sampled_local_class_center : SampledLocalClassCenter}

486
- op : clip
487 488 489 490 491 492 493 494 495 496 497 498
  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
499 500 501
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

502 503 504 505 506 507
- op : clip_by_norm
  inputs :
    x : X
  outputs :
    out : Out

508 509 510 511 512 513 514 515
- op : coalesce_tensor
  inputs :
    {input : Input}
  outputs :
    {output : Output, fused_output : FusedOutput}
  attrs :
    {size_of_dtype : user_defined_size_of_dtype}

516 517 518 519 520 521 522
- op : complex
  backward : complex_grad
  inputs :
    {real : X, imag : Y}
  outputs :
    out : Out

523
- op : concat
L
lzydev 已提交
524
  backward : concat_grad, concat_double_grad
525 526 527 528 529 530 531 532 533 534
  inputs:
    x: X
  outputs:
    out: Out
  attrs:
    axis: axis
  scalar :
    axis :
      data_type : int
      tensor_name : AxisTensor
L
lzydev 已提交
535
  drop_empty_grad : [x_grad]
536 537
  extra :
    attrs : [bool use_mkldnn = false, bool use_quantizer = false, str mkldnn_data_type = "float32"]
L
lzydev 已提交
538 539
  get_expected_kernel_type :
    concat : GetConcatExpectedKernelType
540

541 542 543 544 545
- op : conditional_block
  backward : conditional_block_grad
  extra :
    attrs : ['str[] skip_eager_deletion_vars = {}']

546 547 548 549 550 551
- op : conj
  inputs :
    x : X
  outputs :
    out : Out

552
- op : conv2d
553 554 555 556 557
  backward : conv2d_grad, conv2d_grad_grad
  inputs :
    {input : Input, filter : Filter}
  outputs :
    out : Output
558
  extra :
559
    attrs : [bool is_test = false, bool use_cudnn = true, bool use_mkldnn = false, bool use_addto = false,
560
             bool force_fp32_output = false,
561
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
562 563
  get_expected_kernel_type :
    conv2d : GetConvExpectedKernelType
564

565
- op : conv2d_fusion
F
Feiyu Chan 已提交
566
  extra :
567
    attrs : [bool is_test = false, bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
F
Feiyu Chan 已提交
568
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
569
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
F
Feiyu Chan 已提交
570 571
             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,
572
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
573

574
- op : conv2d_transpose
575 576 577 578 579 580 581 582 583
  backward : conv2d_transpose_grad, conv2d_transpose_double_grad (conv2d_transpose_grad_grad)
  inputs :
    {x : Input, filter : Filter, bias : Bias}
  outputs :
    out : Output
  int_array :
    output_size :
      data_type : int
      support_tensor : true
584
  extra :
585
    inputs : [bias]
586 587 588
    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,
589
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
590

591
- op : conv3d
592 593 594 595 596
  backward : conv3d_grad, conv3d_double_grad (conv3d_grad_grad)
  inputs :
    {input : Input, filter : Filter}
  outputs :
    out : Output
597 598 599 600
  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,
601
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
602 603
  get_expected_kernel_type :
    conv3d : GetConvExpectedKernelType
604

605
- op : conv3d_transpose
606
  backward : conv3d_transpose_grad
607 608 609 610
  inputs :
    {x : Input, filter : Filter}
  outputs :
    out : Output
611
  extra :
612
    attrs : [bool use_cudnn = true, bool use_mkldnn = false, int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
F
Feiyu Chan 已提交
613

614
- op : cos
615
  backward : cos_grad, cos_double_grad, cos_triple_grad
616 617 618 619
  inputs :
    x : X
  outputs :
    out : Out
620 621 622
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

623
- op : cosh
624
  backward : cosh_grad
625 626 627 628
  inputs :
    x : X
  outputs :
    out : Out
629 630 631
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647
- 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

648
- op : cross
649 650
  inputs :
    {x : X, y : Y}
651 652 653 654 655
  attrs :
    axis : dim
  outputs :
    out : Out

656 657 658 659 660 661 662
- 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}

663 664 665 666 667 668 669 670 671
- op : cumprod
  backward : cumprod_grad
  inputs :
    x : X
  attrs :
    dim : dim
  outputs :
    out : Out

G
GGBond8488 已提交
672 673 674 675 676 677 678 679 680
- op : cumsum
  backward: cumsum_grad
  inputs :
    x : X
  outputs :
    out : Out
  scalar:
    axis:
      data_type : int
L
lzydev 已提交
681
      support_tensor : true
G
GGBond8488 已提交
682

683
- op : data_norm
684 685 686 687
  backward : data_norm_grad
  extra :
    attrs : [bool use_mkldnn = false]

688 689 690 691 692 693
- op : decode_jpeg
  inputs :
    x : X
  outputs :
    out : Out

694 695 696 697 698 699 700
- op : deformable_conv
  backward : deformable_conv_grad
  inputs :
    {x : Input, offset : Offset, filter : Filter, mask : Mask}
  outputs :
    out : Output

701
- op : depthwise_conv2d
702 703 704 705 706
  backward : depthwise_conv2d_grad, depthwise_conv2d_double_grad (depthwise_conv2d_grad_grad)
  inputs :
    {input : Input, filter : Filter}
  outputs :
    out : Output
707
  extra :
708
    attrs : [bool is_test = false, bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
709 710 711 712
             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,
713
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB(), bool exhaustive_search = false]
714 715
  get_expected_kernel_type :
    depthwise_conv2d : GetConvExpectedKernelType
716

717
- op : depthwise_conv2d_transpose
718
  backward : depthwise_conv2d_transpose_grad
719 720 721 722 723 724 725 726
  inputs :
    {x : Input, filter : Filter, bias: Bias}
  outputs :
    out : Output
  int_array :
    output_size :
      data_type : int
      support_tensor : true
727
  extra :
728
    inputs : [bias]
729 730 731
    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,
732
             int workspace_size_MB = phi::backends::gpu::GetDefaultConvWorkspaceSizeLimitMB()]
733

734 735 736 737
- op : dequantize_linear
  extra :
    attrs : [float moving_rate = 0.9]

738 739 740 741 742 743 744
- op : det (determinant)
  backward : det_grad (determinant_grad)
  inputs :
    x : Input
  outputs :
    out : Out

745
- op : diag (diag_v2)
746
  backward : diag_grad (diag_v2_grad)
747 748 749 750 751
  inputs :
    x : X
  outputs :
    out : Out

752 753 754 755 756 757
- op : diag_embed
  inputs :
    input : Input
  outputs :
    out : Out

758
- op : diagonal
759 760 761 762 763
  inputs :
    x : Input
  outputs :
    out : Out

764
- op : digamma
765 766 767 768 769
  inputs :
    x : X
  outputs :
    out : Out

770 771 772 773 774 775
- op : dirichlet
  inputs :
    alpha : Alpha
  outputs :
    out : Out

776
- op : dist
777 778 779 780 781
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

782 783 784 785
- op : distributed_push_sparse
  extra :
    attrs : ['int[] slots = {}']

786
- op : divide (elementwise_div)
787
  backward : divide_grad (elementwise_div_grad)
788 789 790 791
  inputs :
    {x: X, y : Y}
  outputs :
    out: Out
792
  extra :
793
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
794 795
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

796
- op : dot
797 798 799 800 801
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

802
- op : dropout
803
  backward : dropout_grad
C
cyber-pioneer 已提交
804 805 806 807 808 809 810 811 812 813 814
  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
815 816 817
  extra :
    attrs : [bool fix_seed = false, int seed = 0]

818
- op : dropout_nd
819 820 821 822
  backward : dropout_nd_grad
  extra :
    attrs : [bool fix_seed = false, int seed = 0]

823 824 825 826 827 828 829 830 831 832
- op : edit_distance
  inputs :
    hyps : Hyps
    refs : Refs
    hypslength : HypsLength
    refslength : RefsLength
  outputs :
    sequencenum : SequenceNum
    out : Out

833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852
- 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

853 854 855 856 857 858 859 860 861
- op : eigvalsh
  backward : eigvalsh_grad
  inputs :
    {x : X}
  outputs :
    {eigenvalues : Eigenvalues, eigenvectors : Eigenvectors}
  attrs :
    uplo : UPLO

862 863 864 865 866 867 868 869 870 871
- op : einsum
  backward : einsum_grad
  inputs :
    x : Operands
  outputs:
    {out : Out, inner_cache: InnerCache, xshape : XShape}
  drop_empty_grad: [x_grad]
  extra:
    outputs: [inner_cache, xshape]

872 873
- op : elementwise_pow
  backward : elementwise_pow_grad
874 875 876 877
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
878
  extra :
879
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
880
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
881 882
  complex_promote : [X, Y]
  manual_signature : [elementwise_pow]
883

884
- op : elu
885 886 887 888 889
  backward : elu_grad, elu_double_grad (elu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
890 891 892
  extra :
    attrs : [bool use_mkldnn = false]

893 894
- op : embedding (lookup_table_v2)
  backward : embedding_grad (lookup_table_v2_grad)
895 896 897 898
  inputs :
    {x : Ids, weight : W}
  outputs :
    out : Out
H
hong 已提交
899 900
  attrs :
   sparse : is_sparse
901
  manual_signature : [embedding_grad]
902 903 904 905 906
  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 = {}']

907 908 909 910 911 912 913 914 915
- op : empty
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList

916 917 918 919 920 921
- op : equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

922 923 924 925 926 927
- op : equal_all
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

928
- op : erf
929 930 931 932 933
  inputs :
    x : X
  outputs :
    out : Out

934
- op : erfinv
935 936 937 938 939
  inputs :
    x : X
  outputs :
    out : Out

940
- op : exp
941
  backward : exp_grad
942 943 944 945
  inputs :
    x : X
  outputs :
    out : Out
946 947
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
948

949
- op : expand (expand_v2)
950
  backward : expand_grad (expand_v2_grad), expand_double_grad(expand_v2_double_grad)
951 952 953 954 955 956 957 958 959 960 961
  inputs :
    x : X
  attrs :
   shape : shape
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int
      tensor_name : Shape
      tensors_name : expand_shapes_tensor
962 963
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
964
  manual_signature : [expand, expand_grad]
965

C
cyber-pioneer 已提交
966
- op : expand_as (expand_as_v2)
967
  backward : expand_as_grad (expand_as_v2_grad)
C
cyber-pioneer 已提交
968
  inputs :
969
    {x : X, y : Y}
C
cyber-pioneer 已提交
970 971 972
  outputs :
    out : Out

973
- op : expm1
974
  backward : expm1_grad
975 976 977 978
  inputs :
    x : X
  outputs :
    out : Out
979 980 981
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

982 983 984 985 986 987 988 989 990
- op : exponential_
  backward : exponential__grad
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    lam : lambda

991 992 993 994 995 996 997 998 999
- op : eye
  outputs :
    out : Out
  scalar :
    num_rows :
      support_tensor : true
    num_columns :
      support_tensor : true

1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027
- 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]

1028 1029 1030
- op : feed
  outputs: {out: Out}

1031
- op : fft_c2c
1032 1033 1034
  inputs: {x: X}
  outputs: {out: Out}

1035
- op : fft_c2r
1036 1037 1038
  inputs: {x: X}
  outputs: {out: Out}

1039
- op : fft_r2c
1040 1041 1042
  inputs: {x: X}
  outputs: {out: Out}

1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053
- op : fill (fill_any)
  backward : fill_grad (fill_any_grad)
  inputs :
    x : X
  outputs :
    out : Out
  scalar :
    value :
      data_type : float
      support_tensor : true

1054 1055 1056 1057 1058 1059 1060
- op : fill_diagonal
  backward : fill_diagonal_grad
  inputs :
    x : X
  outputs :
    out : Out

1061 1062 1063 1064 1065 1066
- op : fill_diagonal_tensor
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

C
cyber-pioneer 已提交
1067
- op : flatten (flatten_contiguous_range)
1068
  backward : flatten_grad (flatten_contiguous_range_grad)
C
cyber-pioneer 已提交
1069 1070 1071 1072 1073 1074
  inputs :
    x : X
  outputs :
    {out : Out, xshape : XShape}
  attrs :
    {start_axis : start_axis, stop_axis : stop_axis}
1075 1076 1077
  extra :
    outputs : [xshape]
  manual_signature : [flatten, flatten_grad]
C
cyber-pioneer 已提交
1078

1079 1080 1081 1082 1083 1084
- op : flip
  inputs :
    x : X
  outputs :
    out : Out

1085 1086
- op : floor
  backward : floor_grad
1087 1088 1089 1090
  inputs :
    x : X
  outputs :
    out : Out
1091 1092 1093
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1094
- op : floor_divide (elementwise_floordiv)
1095 1096 1097 1098
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1099
  extra :
1100
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1101
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
1102 1103
  complex_promote : [X, Y]
  manual_signature : [floor_divide]
1104 1105 1106

- op : fmax (elementwise_fmax)
  backward : fmax_grad (elementwise_fmax_grad)
1107 1108 1109 1110
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1111
  extra :
1112
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1113
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
1114 1115
  complex_promote : [X, Y]
  manual_signature : [fmax]
1116 1117 1118

- op : fmin (elementwise_fmin)
  backward : fmin_grad (elementwise_fmin_grad)
1119 1120 1121 1122
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1123
  extra :
1124
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1125
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
1126 1127
  complex_promote : [X, Y]
  manual_signature : [fmin]
1128

1129 1130 1131 1132 1133 1134
- op : fold
  inputs :
    x : X
  outputs :
    out : Y

1135 1136 1137 1138 1139 1140 1141
- op : frame
  backward : frame_grad
  inputs :
    x : X
  outputs :
    out : Out

1142
- op : frobenius_norm
1143
  backward : frobenius_norm_grad
1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156
  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
1157 1158 1159
  extra :
    attrs : [bool use_mkldnn = false]

1160
- op : full (fill_constant)
1161 1162
  outputs :
    out : Out
1163 1164 1165
  extra :
    attrs : [bool use_mkldnn = false]

C
cyber-pioneer 已提交
1166 1167 1168 1169 1170
- op : full_like (fill_any_like)
  inputs :
    x : X
  outputs :
    out : Out
1171 1172 1173 1174
  scalar :
    value :
      data_type : float
      support_tensor : true
C
cyber-pioneer 已提交
1175

1176 1177 1178 1179 1180
- 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}']

1181 1182 1183 1184
- op : fused_transpose
  extra :
    attrs : [str data_format = "AnyLayout"]

1185 1186
- op : gather
  backward : gather_grad
1187 1188 1189 1190 1191 1192 1193 1194
  inputs :
    {x : X, index : Index}
  outputs :
    out : Out
  scalar :
    axis :
      data_type : int
      tensor_name : Axis
1195

1196 1197 1198 1199 1200 1201 1202
- op : gather_nd
  backward : gather_nd_grad
  inputs :
    {x : X, index : Index}
  outputs :
    out : Out

1203 1204 1205 1206 1207 1208
- op : gather_tree
  inputs :
    {ids : Ids, parents : Parents}
  outputs :
    out : Out

1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220
- 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]

1221
- op : gelu
1222
  backward : gelu_grad
1223 1224 1225 1226
  inputs :
    x : X
  outputs :
    out : Out
1227
  extra :
H
HongyuJia 已提交
1228
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
1229

1230 1231 1232 1233 1234 1235 1236 1237
- op : generate_proposals(generate_proposals_v2)
  inputs :
    {scores : Scores, bbox_deltas : BboxDeltas, im_shape : ImShape, anchors : Anchors, variances : Variances}
  outputs :
    {rpn_rois : RpnRois, rpn_roi_probs : RpnRoiProbs, rpn_rois_num : RpnRoisNum}
  attrs :
    {pre_nms_top_n : pre_nms_topN, post_nms_top_n : post_nms_topN}

1238
- op : grad_add
1239 1240 1241 1242
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1243
  extra :
1244
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1245 1246
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258
- op : greater_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

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

1259 1260 1261 1262 1263 1264
- op : grid_sample(grid_sampler)
  backward : grid_sample_grad (grid_sampler_grad)
  inputs :
    {x : X, grid : Grid}
  outputs :
    out : Output
1265 1266 1267
  extra :
    attrs : [bool use_cudnn = true]

C
cyber-pioneer 已提交
1268 1269 1270 1271 1272 1273 1274 1275 1276 1277
- op : group_norm
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    y : Y
    mean : Mean
    variance : Variance

1278
- op : gru
1279 1280 1281 1282
  backward : gru_grad
  extra :
    attrs : [bool is_test = false]

1283 1284 1285 1286 1287 1288
- op : gumbel_softmax
  inputs :
    x : X
  outputs :
    out : Out

1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302
- 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 已提交
1303 1304 1305 1306 1307
- op : hardswish (hard_swish)
  inputs :
    x : X
  outputs :
    out : Out
1308
  backward : hardswish_grad (hard_swish_grad)
C
cyber-pioneer 已提交
1309 1310
  extra :
    attrs : [bool use_mkldnn = false]
1311
  manual_signature : [hardswish]
C
cyber-pioneer 已提交
1312

1313 1314 1315 1316 1317 1318 1319
- op : hardtanh (brelu)
  backward : hardtanh_grad (brelu_grad)
  inputs :
    x : X
  outputs :
    out : Out

1320 1321
- op : heaviside (elementwise_heaviside)
  backward : heaviside_grad (elementwise_heaviside_grad)
1322 1323 1324 1325
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1326
  extra :
1327
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1328
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
1329
  complex_promote : [X, Y]
1330

1331 1332 1333 1334 1335 1336
- op : histogram
  inputs :
    input : X
  outputs :
    out : Out

K
kangguangli 已提交
1337 1338 1339 1340 1341 1342 1343
- op : hsigmoid_loss(hierarchical_sigmoid)
  backward: hsigmoid_loss_grad(hierarchical_sigmoid_grad)
  inputs:
   {x: X, w: W, label: Label, bias: Bias, path: PathTable, code: PathCode}
  outputs:
   {out: Out, pre_out: PreOut, w_out: W_Out}

1344 1345 1346 1347 1348 1349 1350
- op : huber_loss
  backward : huber_loss_grad
  inputs :
    {input : X, label : Y}
  outputs :
    {out : Out, residual : Residual}

1351 1352 1353 1354 1355 1356 1357
- op : imag
  backward : imag_grad
  inputs :
    x : X
  outputs :
    out : Out

1358 1359 1360 1361 1362 1363
- op : increment
  inputs :
    x : X
  outputs :
    out : Out

1364 1365 1366 1367 1368 1369
- op : index_add
  inputs :
    {x : X, index : Index, add_value : AddValue}
  outputs :
    out : Out

1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383
- 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

1384
- op : inplace_abn
1385 1386 1387 1388
  backward : inplace_abn_grad
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

C
cyber-pioneer 已提交
1389 1390 1391 1392 1393 1394 1395 1396 1397
- op : instance_norm
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    y : Y
    saved_mean : SavedMean
    saved_variance : SavedVariance
1398 1399 1400 1401
  extra:
    outputs: [ saved_mean, saved_variance ]
  get_expected_kernel_type:
    instance_norm: GetInstanceNormExpectedKernelType
C
cyber-pioneer 已提交
1402

1403 1404 1405 1406 1407 1408
- op : inverse
  inputs :
    x : Input
  outputs :
    out : Output

1409 1410 1411 1412 1413 1414
- op : is_empty
  inputs :
    x : X
  outputs :
    out : Out

1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427
- 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

1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445
- 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

1446 1447 1448 1449 1450 1451 1452
- op : kldiv_loss
  backward : kldiv_loss_grad
  inputs :
    {x : X, label : Target}
  outputs :
    out : Loss

1453 1454 1455 1456 1457 1458 1459 1460
- op : kron
  backward : kron_grad
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
  complex_promote : [X, Y]

1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472
- op : kthvalue
  inputs :
    x : X
  outputs :
    {out : Out, indices : Indices}

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

H
hong 已提交
1473
- op : lamb_ (lamb)
1474 1475 1476 1477 1478
  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}

1479
- op : layer_norm
1480
  backward : layer_norm_grad
C
cyber-pioneer 已提交
1481 1482 1483 1484 1485 1486 1487 1488
  inputs :
    x : X
    scale : Scale
    bias : Bias
  outputs :
    out : Y
    mean : Mean
    variance : Variance
1489 1490
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]
1491 1492
  get_expected_kernel_type :
    layer_norm : GetLayerNormExpectedKernelType
1493

1494
- op : leaky_relu
1495 1496 1497 1498 1499 1500 1501
  backward : leaky_relu_grad, leaky_relu_double_grad (leaky_relu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
  attrs:
    negative_slope : alpha
1502 1503 1504
  extra :
    attrs : [bool use_mkldnn = false]

1505 1506 1507 1508 1509 1510 1511
- op : lerp
  backward : lerp_grad
  inputs :
    {x : X, y : Y, weight : Weight}
  outputs :
    out : Out

1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523
- op : less_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

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

1524
- op : lgamma
1525 1526 1527 1528 1529
  inputs :
    x : X
  outputs :
    out : Out

1530
- op : linear_interp (linear_interp_v2)
1531
  backward : linear_interp_grad (linear_interp_v2_grad)
1532 1533 1534 1535
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
1536 1537 1538
  extra :
    attrs : [bool use_mkldnn = false]

1539 1540 1541 1542 1543 1544
- op : linspace
  inputs :
    {start : Start, stop : Stop, number : Num}
  outputs :
    out : Out

1545
- op : log
1546 1547 1548 1549 1550
  backward : log_grad, log_double_grad (log_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
1551 1552 1553
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1554
- op : log10
1555
  backward : log10_grad
1556 1557 1558 1559
  inputs :
    x : X
  outputs :
    out : Out
1560 1561 1562
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1563
- op : log1p
1564
  backward : log1p_grad
1565 1566 1567 1568
  inputs :
    x : X
  outputs :
    out : Out
1569 1570 1571
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1572
- op : log2
1573
  backward : log2_grad
1574 1575 1576 1577
  inputs :
    x : X
  outputs :
    out : Out
1578 1579 1580
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1581 1582 1583 1584 1585 1586 1587
- op : log_loss
  backward : log_loss_grad
  inputs :
    {input : Predicted, label : Labels}
  outputs :
    out : Loss

1588
- op : log_softmax
1589
  backward : log_softmax_grad
1590 1591 1592 1593
  inputs :
    x : X
  outputs :
    out: Out
1594 1595 1596
  extra :
    attrs : [bool use_mkldnn = false]

1597 1598 1599 1600 1601 1602 1603
- op : logcumsumexp
  backward : logcumsumexp_grad
  inputs :
    x : X
  outputs :
    out : Out

1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627
- 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

1628 1629 1630 1631 1632 1633 1634
- op : logit
  inputs :
    x : X
  outputs :
    out : Out

- op : logsigmoid
1635
  backward : logsigmoid_grad
1636 1637 1638 1639
  inputs :
    x : X
  outputs :
    out : Out
1640 1641 1642
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

1643 1644 1645 1646 1647 1648 1649
- op : logsumexp
  backward : logsumexp_grad
  inputs :
    x : X
  outputs :
    out : Out

1650
- op : lrn
1651 1652 1653 1654
  backward : lrn_grad
  extra :
    attrs : [bool use_mkldnn = false, bool is_test = false]

1655 1656 1657 1658 1659 1660 1661 1662 1663 1664
- op : lstsq
  inputs :
    {x : X, y : Y}
  outputs :
    {solution : Solution, residuals : Residuals, rank : Rank, singular_values : SingularValues}
  scalar :
    rcond :
      data_type : float
      support_tensor : true

1665 1666 1667 1668 1669 1670 1671
- op : lu_unpack
  backward : lu_unpack_grad
  inputs :
    {x : X, y : Pivots}
  outputs :
    {pmat : Pmat, l : L, u : U}

1672 1673 1674 1675 1676 1677 1678
- op : margin_cross_entropy
  backward : margin_cross_entropy_grad
  inputs:
    {logits : Logits, label : Label}
  outputs:
    {softmax : Softmax, loss : Loss}

1679 1680 1681 1682 1683 1684
- op : masked_select
  inputs :
    {x : X, mask : Mask}
  outputs :
    out : Y

1685
- op : matmul (matmul_v2)
1686
  backward : matmul_grad (matmul_v2_grad), matmul_double_grad (matmul_v2_grad_grad), matmul_triple_grad (matmul_v2_triple_grad)
1687 1688 1689 1690 1691 1692
  inputs :
    {x : X, y : Y}
  attrs :
    {transpose_x : trans_x, transpose_y : trans_y}
  outputs :
    out : Out
1693
  extra :
1694
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
1695
  complex_promote : [X, Y]
1696

1697 1698 1699 1700 1701 1702
- 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]

1703 1704 1705 1706 1707 1708 1709 1710
- op : matrix_nms
  inputs :
    {bboxes : BBoxes, scores : Scores}
  outputs :
    {out : Out, index : Index, roisnum : RoisNum}
  get_expected_kernel_type :
    matrix_nms : GetMatrixNmsExpectedKernelType

1711 1712 1713 1714 1715 1716
- op : matrix_power
  inputs :
    x : X
  outputs :
    out : Out

1717 1718 1719 1720 1721 1722 1723
- op : matrix_rank
  inputs :
    {x : X, tol_tensor : TolTensor}
  outputs :
    out : Out
  manual_signature : [matrix_rank]

1724 1725 1726 1727 1728 1729 1730 1731
- op : max (reduce_max)
  backward : max_grad (reduce_max_grad)
  inputs:
    x : X
  attrs:
    { axis : dim,  keepdim : keep_dim}
  outputs:
    out : Out
1732 1733
  extra :
    attrs : [bool use_mkldnn = false]
1734 1735 1736
  int_array:
    axis :
      data_type : int
1737 1738 1739 1740 1741
      support_tensor : true
  get_expected_kernel_type :
    max : GetReduceExpectedKernelType
    max_grad : GetReduceGradExpectedKernelType
  manual_signature : [max]
1742

1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758
- op : max_pool2d_with_index
  inputs :
    {x : X}
  outputs :
    {out : Out, mask : Mask}
  attrs :
    kernel_size : ksize

- op : max_pool3d_with_index
  inputs :
    {x : X}
  outputs :
    {out : Out, mask : Mask}
  attrs :
    kernel_size : ksize

1759 1760
- op : maximum (elementwise_max)
  backward : maximum_grad (elementwise_max_grad)
1761 1762 1763 1764
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1765
  extra :
1766
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1767
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
1768 1769
  complex_promote : [X, Y]
  manual_signature : [maximum]
1770

1771 1772 1773 1774 1775 1776
- op : maxout
  inputs :
    x : X
  outputs :
    out : Out

C
cyber-pioneer 已提交
1777
- op : mean (reduce_mean)
K
kangguangli 已提交
1778
  backward : mean_grad (reduce_mean_grad)
C
cyber-pioneer 已提交
1779 1780 1781 1782 1783 1784 1785 1786 1787
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    {axis : dim, keepdim : keep_dim}
  extra :
    attrs : [bool use_mkldnn = false]

1788 1789 1790 1791 1792 1793 1794
- op : mean_all (mean)
  backward : mean_all_grad (mean_grad)
  inputs :
    x : X
  outputs :
    out : Out

1795 1796 1797 1798 1799 1800
- op : merge_selected_rows
  inputs :
    x : X
  outputs :
    out : Out

1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816
- op : merged_adam_
  inputs :
    {param: Param, grad: Grad, learning_rate: LearningRate, moment1: Moment1, moment2: Moment2, beta1_pow: Beta1Pow, beta2_pow: Beta2Pow, master_param: MasterParam}
  outputs :
    {param_out: ParamOut, moment1_out: Moment1Out, moment2_out: Moment2Out, beta1_pow_out: Beta1PowOut, beta2_pow_out: Beta2PowOut, master_param_out: MasterParamOut}
  scalar :
    beta1 :
      data_type : float
      support_tensor : true
    beta2 :
      data_type : float
      support_tensor : true
    epsilon :
      data_type : float
      support_tensor : true

1817
- op : merged_momentum_ (merged_momentum)
1818 1819 1820 1821 1822
  inputs :
    {param : Param, grad : Grad, velocity : Velocity, learning_rate : LearningRate, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, velocity_out : VelocityOut, master_param_out : MasterParamOut}

1823 1824 1825 1826 1827 1828 1829 1830
- op : meshgrid
  backward : meshgrid_grad
  inputs :
    inputs : X
  outputs :
    out : Out
  drop_empty_grad : [inputs_grad]

1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849
- op : min (reduce_min)
  backward : min_grad (reduce_min_grad)
  inputs:
    x : X
  outputs:
    out : Out
  attrs:
    { axis : dim,  keepdim : keep_dim}
  extra :
    attrs : [bool use_mkldnn = false]
  int_array:
    axis :
      data_type : int
      support_tensor : true
  get_expected_kernel_type :
    min : GetReduceExpectedKernelType
    min_grad : GetReduceGradExpectedKernelType
  manual_signature : [min]

1850 1851
- op : minimum (elementwise_min)
  backward : minimum_grad (elementwise_min_grad)
1852 1853 1854 1855
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
1856 1857 1858
  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]
1859 1860
  complex_promote : [X, Y]
  manual_signature : [minimum]
1861

1862 1863
- op : mish
  backward : mish_grad
1864
  inputs:
1865 1866 1867
    {x : X, lambda : threshold}
  outputs:
    out: Out
1868 1869 1870
  extra :
    attrs : [bool use_mkldnn = false]

1871 1872 1873 1874 1875 1876 1877
- op : mode
  backward : mode_grad
  inputs :
    x : X
  outputs :
    {out : Out, indices : Indices}

H
hong 已提交
1878
- op : momentum_ (momentum)
1879 1880 1881 1882 1883
  inputs :
    {param : Param, grad : Grad, velocity : Velocity, learning_rate : LearningRate, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, velocity_out : VelocityOut, master_param_out : MasterParamOut}

1884 1885 1886 1887 1888 1889 1890 1891
- op : multi_dot
  backward : multi_dot_grad
  inputs :
    x : X
  outputs :
    out : Out
  drop_empty_grad : [x_grad]

1892 1893 1894 1895 1896 1897
- op : multiclass_nms3
  inputs :
    {bboxes : BBoxes, scores : Scores, rois_num : RoisNum}
  outputs :
    {out : Out, index : Index, nms_rois_num : NmsRoisNum}

1898 1899 1900 1901 1902 1903 1904 1905 1906 1907
- op : multinomial
  inputs :
    {x : X}
  outputs :
    out : Out
  scalar :
    num_samples :
      data_type : int
      support_tensor : true

1908 1909 1910 1911 1912 1913 1914 1915
- op : multiplex
  backward : multiplex_grad
  inputs :
    {inputs : X, index : Ids}
  outputs :
    out : Out
  drop_empty_grad : [inputs_grad]

1916 1917
- op : multiply (elementwise_mul)
  backward : multiply_grad (elementwise_mul_grad)
1918 1919 1920 1921
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out
1922
  extra :
1923
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
1924 1925
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

1926
- op : mv
1927 1928 1929 1930 1931
  inputs :
    {x : X, vec : Vec}
  outputs :
    out : Out

Z
zyfncg 已提交
1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943
- op : nanmedian
  backward : nanmedian_grad
  inputs :
    {x : X}
  outputs :
    {out : Out, medians : MedianIndex}
  int_array:
    axis:
      data_type : int
  extra:
    outputs : [medians]

1944 1945 1946 1947 1948 1949
- op : nce
  backward : nce_grad
  extra :
    attrs : [int trainer_id = 0, 'int64_t[] height_sections = {}', 'str[] epmap = {}',
             'str[] table_names = {}', 'int[] custom_neg_classes = {}']

1950
- op : nearest_interp (nearest_interp_v2)
1951
  backward : nearest_interp_grad (nearest_interp_v2_grad)
1952 1953 1954 1955
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
1956 1957 1958
  extra :
    attrs : [bool use_mkldnn = false]

1959 1960 1961 1962 1963 1964 1965
- op : nll_loss
  backward : nll_loss_grad
  inputs :
    {input : X, label : Label, weight : Weight}
  outputs :
    {out : Out, total_weight : Total_weight}

1966 1967 1968 1969 1970 1971 1972 1973
- op : nms
  inputs :
    x : Boxes
  outputs :
    out : KeepBoxesIdxs
  attrs :
    threshold : iou_threshold

C
cyber-pioneer 已提交
1974 1975 1976 1977 1978 1979
- op : nonzero (where_index)
  inputs :
    condition : Condition
  outputs :
    out : Out

L
lzydev 已提交
1980 1981 1982 1983 1984 1985 1986 1987 1988
- op : norm
  backward : norm_grad
  inputs :
    x : X
  outputs :
    {out : Out, norm : Norm}
  extra :
    outputs : [norm]

1989 1990 1991 1992 1993 1994
- op : not_equal
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

1995 1996 1997 1998 1999 2000
- op : numel(size)
  inputs :
    x : Input
  outputs :
    size : Out

2001 2002 2003 2004 2005 2006 2007 2008 2009 2010
- op : one_hot (one_hot_v2)
  inputs :
    x : X
  outputs :
    out : Out
  scalar :
    depth :
      data_type : int
      tensor_name : depth_tensor

2011 2012 2013 2014 2015 2016 2017
- op : overlap_add
  backward : overlap_add_grad
  inputs :
    x : X
  outputs :
    out : Out

2018 2019 2020 2021 2022 2023 2024
- op : p_norm
  backward: p_norm_grad
  inputs :
    x : X
  outputs :
    out : Out

2025
- op : pad2d
2026 2027 2028 2029
  backward : pad2d_grad
  extra :
    attrs : [bool use_mkldnn = false]

2030
- op : pad3d
2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041
  backward : pad3d_grad, pad3d_double_grad
  inputs :
    x : X
  outputs :
    out : Out
  int_array:
    paddings :
      data_type : int
      tensor_name : Paddings
  attrs :
    pad_value : value
2042 2043 2044
  extra :
    attrs : [bool use_mkldnn = false]

2045
- op : partial_sum
2046 2047 2048 2049
  backward : partial_sum_grad
  extra :
    attrs : [bool use_mkldnn = false]

2050 2051 2052 2053 2054 2055 2056
- op : pixel_shuffle
  backward : pixel_shuffle_grad
  inputs :
    x : X
  outputs :
    out : Out

2057
- op : poisson
2058 2059 2060 2061 2062
  inputs :
    x : X
  outputs :
    out : Out

2063
- op : pool2d
2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078
  backward : pool2d_grad, pool2d_double_grad
  inputs :
    {x : X}
  outputs :
    {out : Out}
  attrs :
    {kernel_size : ksize}
  int_array:
    kernel_size :
      data_type : int
      support_tensor : true
  get_expected_kernel_type :
    pool2d : GetPoolExpectedKernelType
    pool2d_grad : GetPoolExpectedKernelType
    pool2d_double_grad : GetPoolDoubleGradExpectedKernelType
2079 2080
  extra :
    attrs : [bool use_mkldnn = false, bool use_quantizer = false,
2081
              str mkldnn_data_type = "float32", bool is_test = false]
2082 2083 2084

- op : pool3d
  backward : pool3d_grad
2085 2086 2087 2088 2089 2090 2091 2092 2093
  inputs :
    {x : X}
  outputs :
    {out : Out}
  attrs :
    {kernel_size : ksize}
  get_expected_kernel_type :
    pool3d : GetPoolExpectedKernelType
    pool3d_grad : GetPoolExpectedKernelType
2094
  extra :
2095
    attrs : [bool use_mkldnn = false]
2096

2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109
- 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

2110
- op : prelu
2111
  backward : prelu_grad
2112 2113 2114 2115
  inputs :
    { x : X, alpha : Alpha}
  outputs :
    out : Out
2116 2117 2118
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]

2119 2120 2121 2122 2123 2124
- op : print
  inputs :
    in : In
  outputs :
    out : Out

2125 2126 2127 2128 2129 2130 2131 2132
- op : prior_box
  inputs :
    {input: Input, image: Image}
  outputs :
    {out: Boxes, var: Variances}
  extra :
    attrs : [bool use_mkldnn = false, bool use_quantizer = false, str mkldnn_data_type = "float32"]

2133 2134 2135 2136 2137 2138
- op : prod (reduce_prod)
  backward : prod_grad (reduce_prod_grad)
  inputs:
    x : X
  outputs:
    out : Out
2139 2140
  attrs:
    { dims : dim,  keep_dim : keep_dim}
2141
  int_array:
2142
    dims :
2143
      data_type : int
2144
      support_tensor : true
2145 2146
  extra :
    attrs : [bool use_mkldnn = false]
2147 2148 2149 2150
  get_expected_kernel_type :
    prod : GetReduceExpectedKernelType
    prod_grad : GetReduceGradExpectedKernelType
  manual_signature : [prod]
2151

2152 2153 2154 2155 2156 2157 2158
- op : psroi_pool
  backward : psroi_pool_grad
  inputs :
    {x : X, boxes : ROIs, boxes_num : RoisNum}
  outputs :
    out : Out

2159 2160 2161 2162 2163 2164 2165 2166 2167
- op : put_along_axis
  backward : put_along_axis_grad
  inputs :
    {arr : Input, indices : Index, values : Value}
  outputs :
    out : Result
  attrs :
    {axis : Axis, reduce : Reduce}

2168 2169 2170 2171 2172 2173 2174
- op : qr
  backward : qr_grad
  inputs :
    x : X
  outputs :
    {q : Q, r : R}

2175 2176 2177 2178
- op : quantize_linear
  extra :
    attrs : [float moving_rate = 0.9]

2179 2180 2181 2182 2183 2184 2185 2186 2187 2188
- op : randint
  outputs :
    out : Out
  int_array:
    shape :
      data_type : int64_t
      tensor_name : ShapeTensor
      tensors_name : ShapeTensorList
  manual_signature : [randint]

2189 2190 2191 2192 2193 2194
- op : randperm
  outputs :
    out : Out
  extra :
    attrs : [int seed = 0]

2195 2196 2197 2198 2199 2200 2201
- op : real
  backward : real_grad
  inputs :
    x : X
  outputs :
    out : Out

2202
- op : reciprocal
2203
  backward : reciprocal_grad
2204 2205 2206 2207
  inputs :
    x : X
  outputs :
    out : Out
2208 2209 2210
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2211
- op : relu
2212 2213 2214 2215 2216
  backward : relu_grad, relu_double_grad (relu_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2217 2218 2219
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2220
- op : relu6
2221
  backward : relu6_grad
2222 2223 2224 2225
  inputs :
    x : X
  outputs :
    out : Out
2226
  extra :
Z
zhangyuqin1998 已提交
2227
    attrs : [bool use_mkldnn = false, float threshold = 6.0]
2228

2229
- op : remainder (elementwise_mod)
2230 2231 2232 2233
  inputs :
    {x : X, y : Y}
  outputs :
    {out : Out}
2234
  extra :
2235
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
2236
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]
2237 2238
  complex_promote : [X, Y]
  manual_signature : [remainder]
2239

2240
- op : renorm
2241
  backward : renorm_grad
2242 2243 2244 2245
  inputs :
    x : X
  outputs :
    out : Out
2246 2247 2248
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2249 2250 2251 2252 2253 2254 2255 2256
- op : repeat_interleave
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    repeats : Repeats

2257 2258 2259 2260 2261 2262
- op : reshape (reshape2)
  backward : reshape_grad (reshape2_grad)
  inputs:
    x : X
  outputs:
    out : Out
K
kangguangli 已提交
2263
    xshape: XShape
2264 2265 2266 2267 2268 2269 2270 2271
  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]

2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282
- op : reverse
  inputs:
    x : X
  outputs:
    out : Out
  int_array:
    axis :
      data_type : int
      support_tensor : true
  manual_signature : [reverse]

H
hong 已提交
2283
- op : rmsprop_ (rmsprop)
2284 2285 2286 2287 2288
  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}

2289 2290 2291 2292 2293 2294 2295 2296
- op : rnn
  backward : rnn_grad
  inputs:
    { x : Input, pre_state : PreState, weight_list : WeightList, sequence_length : SequenceLength}
  outputs:
    { out : Out, dropout_state_out : DropoutState, state : State, reserve : Reserve}
  drop_empty_grad : [pre_state_grad, weight_list_grad]

2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310
- op : roi_align
  backward : roi_align_grad
  inputs :
    {x : X, boxes : ROIs, boxes_num : RoisNum}
  outputs :
    out : Out

- op : roi_pool
  backward : roi_pool_grad
  inputs :
    {x : X, boxes : ROIs, boxes_num : RoisNum}
  outputs :
    {out : Out, arg_max : Argmax}

2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321
- op : roll
  backward : roll_grad
  inputs :
    x : X
  outputs :
    out : Out
  int_array :
    shifts :
      data_type : int64_t
      tensor_name : ShiftsTensor

2322
- op : round
2323
  backward : round_grad
2324 2325 2326 2327
  inputs :
    x : X
  outputs :
    out : Out
2328
  extra :
2329 2330
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2331
- op : rsqrt
2332 2333 2334 2335 2336
  backward : rsqrt_grad, rsqrt_double_grad (rsqrt_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2337 2338
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
2339

2340
- op : scale
HappyHeavyRain's avatar
HappyHeavyRain 已提交
2341
  backward : scale_grad
2342 2343 2344
  inputs :
    x : X
  outputs :
HappyHeavyRain's avatar
HappyHeavyRain 已提交
2345 2346 2347 2348 2349
    out : Out
  scalar :
    scale :
      data_type : float
      tensor_name : ScaleTensor
2350 2351 2352
  extra :
    attrs : [bool use_mkldnn = false]

2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366
- 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

2367 2368 2369 2370 2371 2372
- op : searchsorted
  inputs :
    {sorted_sequence : SortedSequence, values : Values}
  outputs :
    out : Out

2373
- op : seed
2374 2375 2376
  extra :
    attrs : [bool deterministic = false, str rng_name = "", bool force_cpu = false]

L
lzydev 已提交
2377 2378 2379 2380 2381 2382 2383
- op : segment_pool
  backward : segment_pool_grad
  inputs :
    {x : X, segment_ids : SegmentIds}
  outputs :
    {out : Out, summed_ids : SummedIds}

2384 2385 2386 2387 2388 2389 2390
- op : selu
  backward : selu_grad
  inputs :
    x : X
  outputs :
    out : Out

2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412
- 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

2413 2414 2415
- op : send_uv (graph_send_uv)
  backward : send_uv_grad (graph_send_uv_grad)

2416 2417 2418 2419 2420
- op : sequence_softmax
  backward : sequence_softmax_grad
  extra :
    attrs : [str data_format = "AnyLayout"]

H
hong 已提交
2421
- op : sgd_ (sgd)
2422 2423 2424 2425 2426
  inputs :
    {param : Param, learning_rate : LearningRate, grad : Grad, master_param : MasterParam}
  outputs :
    {param_out : ParamOut, master_param_out : MasterParamOut}
  get_expected_kernel_type :
2427
    sgd_ : GetSgdExpectedKernelType
2428 2429 2430
  extra :
    attrs : [bool use_mkldnn=false]

H
hong 已提交
2431 2432 2433 2434
- op : shadow_output
  inputs: {x: x}
  outputs: {out: out}

2435 2436 2437 2438 2439 2440
- op : shape
  inputs :
    input : Input
  outputs :
    out : Out

2441
- op : shape
2442 2443 2444
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]

2445 2446 2447 2448 2449 2450
- op : shard_index
  inputs :
    input : X
  outputs :
    out : Out

2451 2452 2453 2454 2455 2456 2457
- op : share_buffer
  inputs :
    x : X
  outputs :
    out : Out
    xout : XOut

2458
- op : shuffle_channel
2459 2460 2461 2462
  backward : shuffle_channel_grad
  extra :
    attrs : [bool use_mkldnn = false]

2463
- op : sigmoid
2464 2465 2466 2467 2468
  backward : sigmoid_grad, sigmoid_double_grad (sigmoid_grad_grad), sigmoid_triple_grad
  inputs :
    x : X
  outputs :
    out : Out
2469 2470 2471
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

HappyHeavyRain's avatar
HappyHeavyRain 已提交
2472 2473 2474 2475 2476 2477 2478
- op : sign
  backward : sign_grad
  inputs :
    x : X
  outputs :
    out : Out

2479
- op : silu
X
xiaoguoguo626807 已提交
2480
  backward : silu_grad, silu_double_grad
2481 2482 2483 2484
  inputs :
    x : X
  outputs :
    out : Out
2485 2486 2487
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2488
- op : sin
2489
  backward : sin_grad, sin_double_grad, sin_triple_grad
2490 2491 2492 2493
  inputs :
    x : X
  outputs :
    out : Out
2494 2495 2496
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2497
- op : sinh
2498
  backward : sinh_grad
2499 2500 2501 2502
  inputs :
    x : X
  outputs :
    out : Out
2503 2504 2505
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2506
- op : slice
2507
  backward : slice_grad
2508 2509 2510 2511
  inputs :
    input : Input
  outputs :
    out : Out
2512 2513
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
2514 2515 2516 2517 2518 2519 2520 2521 2522
  int_array :
    starts :
      data_type : int
      tensor_name : StartsTensor
      tensors_name : StartsTensorList
    ends :
      data_type : int
      tensor_name : EndsTensor
      tensors_name : EndsTensorList
2523

2524 2525 2526 2527 2528 2529 2530
- op : slogdet(slogdeterminant)
  backward : slogdet_grad(slogdeterminant_grad)
  inputs :
    x : Input
  outputs :
    out : Out

2531
- op : softmax
2532
  backward : softmax_grad
2533 2534
  inputs :
    x : X
2535 2536 2537 2538 2539
  outputs :
    out : Out
  get_expected_kernel_type :
    softmax : GetSoftmaxExpectedKernelType
    softmax_grad : GetSoftmaxGradExpectedKernelType
2540
  extra :
2541
    attrs : [str data_format = "AnyLayout", bool use_cudnn = true, bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]
2542

2543
- op : softplus
W
will-jl944 已提交
2544
  backward : softplus_grad, softplus_double_grad
2545 2546 2547 2548
  inputs :
    x : X
  outputs :
    out : Out
2549
  extra :
S
Sławomir Siwek 已提交
2550
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
2551

2552 2553 2554 2555 2556 2557 2558 2559 2560
- op : softshrink
  backward : softshrink_grad
  inputs :
    x : X
  outputs :
    out : Out
  attrs :
    threshold : lambda

2561
- op : softsign
2562
  backward : softsign_grad
2563 2564 2565 2566
  inputs :
    x : X
  outputs :
    out : Out
2567 2568
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]
2569

2570
- op : solve
2571 2572 2573 2574 2575
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

2576 2577 2578 2579 2580 2581 2582
- op : spectral_norm
  backward : spectral_norm_grad
  inputs :
    {weight : Weight, u : U, v : V}
  outputs :
    out : Out

W
wangzhen38 已提交
2583
- op : split
2584 2585 2586 2587
  inputs:
    x : X
  outputs:
    out : Out
W
wangzhen38 已提交
2588 2589 2590
  int_array:
      sections :
          data_type : int
2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601
  scalar :
      axis :
        data_type : int
        support_tensor : true

- op : split_with_num
  scalar :
      axis :
        data_type : int
        support_tensor : true
        tensor_name : AxisTensor
W
wangzhen38 已提交
2602

2603
- op : sqrt
2604 2605 2606 2607 2608
  backward : sqrt_grad, sqrt_double_grad (sqrt_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2609 2610 2611
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2612
- op : square
2613 2614 2615 2616 2617
  backward : square_grad, square_double_grad (square_grad_grad)
  inputs :
    x : X
  outputs :
    out : Out
2618 2619 2620
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2621
- op : squeeze (squeeze2)
2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632
  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
2633 2634
  extra :
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
2635
    outputs : [xshape]
2636

2637
- op : stack
2638
  backward : stack_grad
2639 2640 2641 2642
  inputs :
    x : X
  outputs :
    out : Y
2643 2644
  extra :
    attrs : [bool use_mkldnn = false]
2645
  drop_empty_grad : [x_grad]
2646

Z
zyfncg 已提交
2647 2648 2649 2650 2651 2652 2653
- op : stanh
  backward : stanh_grad
  inputs :
    x : X
  outputs :
    out : Out

2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677
- op : strided_slice
  backward : strided_slice_grad
  inputs :
    x : Input
  outputs :
    out : Out
  int_array :
    starts :
      data_type : int
      tensor_name : StartsTensor
      tensors_name : StartsTensorList
    ends :
      data_type : int
      tensor_name : EndsTensor
      tensors_name : EndsTensorList
    strides :
      data_type : int
      tensor_name : StridesTensor
      tensors_name : StridesTensorList
  manual_signature : [strided_slice, strided_slice_grad]
  get_expected_kernel_type :
    strided_slice : GetStridedSliceExpectedKernelType
    strided_slice_grad : GetStridedSliceGradExpectedKernelType

2678 2679
- op : subtract (elementwise_sub)
  backward : subtract_grad (elementwise_sub_grad)
2680 2681 2682 2683
  inputs :
    {x : X, y: Y}
  outputs :
    out : Out
2684
  extra :
2685
    attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32",
2686 2687
             bool use_quantizer = false, float Scale_x = 1.0f, float Scale_y = 1.0f, float Scale_out = 1.0f]

2688
- op : sum (reduce_sum)
2689
  backward : sum_grad (reduce_sum_grad), sum_double_grad
2690 2691 2692 2693
  inputs:
    {x : X}
  outputs:
    out : Out
2694 2695 2696 2697
  attrs:
    { axis : dim,  keepdim : keep_dim, dtype : out_dtype}
  extra :
    attrs : [bool use_mkldnn = false]
2698 2699 2700
  int_array:
      axis :
        data_type : int
2701 2702 2703 2704 2705
        support_tensor : true
  get_expected_kernel_type :
    sum : GetReduceExpectedKernelType
    sum_grad : GetReduceGradExpectedKernelType
  manual_signature : [sum]
2706

2707 2708 2709 2710 2711 2712 2713
- op : svd
  backward : svd_grad
  inputs :
    x : X
  outputs :
    {u : U, s : S, vh : VH}

2714
- op : swish
2715
  backward : swish_grad
2716 2717 2718 2719
  inputs :
    x : X
  outputs :
    out : Out
2720
  extra :
Z
zhangyuqin1998 已提交
2721
    attrs : [bool use_mkldnn = false, float beta = 1.0]
2722

2723
- op : sync_batch_norm
2724 2725 2726
  inputs :
    {x : X, scale : Scale, bias : Bias, mean : Mean, variance : Variance}
  outputs :
W
wanghuancoder 已提交
2727
    {out : Y, mean_out : MeanOut, variance_out : VarianceOut, saved_mean : SavedMean, saved_variance : SavedVariance, reserve_space : ReserveSpace}
2728 2729 2730 2731
  backward : sync_batch_norm_grad
  extra :
    attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]

2732 2733 2734 2735 2736 2737 2738 2739 2740
- op : take_along_axis
  backward : take_along_axis_grad
  inputs :
    {arr : Input, indices : Index}
  outputs :
    out : Result
  attrs :
    axis : Axis

2741
- op : tan
2742
  backward : tan_grad
2743 2744 2745 2746
  inputs :
    x : X
  outputs :
    out : Out
2747 2748 2749
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2750
- op : tanh
X
xiaoguoguo626807 已提交
2751
  backward : tanh_grad, tanh_double_grad (tanh_grad_grad), tanh_triple_grad
2752 2753 2754 2755
  inputs :
    x : X
  outputs :
    out : Out
2756 2757 2758
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2759
- op : tanh_shrink
2760
  backward : tanh_shrink_grad
2761 2762 2763 2764
  inputs :
    x : X
  outputs :
    out : Out
2765 2766 2767
  extra :
    attrs : [bool use_mkldnn = false, bool use_cudnn = false]

2768 2769 2770 2771 2772 2773
- op : thresholded_relu
  inputs :
    x : X
  outputs :
    out : Out

2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
- 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

2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796
- 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

2797
- op : trace
2798 2799 2800 2801
  inputs :
    x : Input
  outputs :
    out : Out
2802

2803 2804
- op : transpose (transpose2)
  backward : transpose_grad (transpose2_grad)
2805 2806 2807 2808
  inputs :
    x : X
  outputs :
    out : Out
J
Jiabin Yang 已提交
2809 2810
  attrs:
    perm : axis
2811
  extra :
J
Jiabin Yang 已提交
2812
    outputs : [XShape]
2813
    attrs : [bool use_mkldnn = false, str data_format = "AnyLayout", str mkldnn_data_type = "float32"]
2814

2815 2816 2817 2818 2819 2820 2821
- op : triangular_solve
  backward : triangular_solve_grad
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

2822 2823 2824 2825 2826 2827 2828
- op : tril_triu
  backward : tril_triu_grad
  inputs :
    {x: X}
  outputs :
    {out : Out}

2829
- op : trilinear_interp (trilinear_interp_v2)
2830
  backward : trilinear_interp_grad (trilinear_interp_v2_grad)
2831 2832 2833 2834
  inputs :
    {x : X, out_size : OutSize, size_tensor : SizeTensor, scale_tensor : Scale}
  outputs :
    output : Out
2835 2836 2837
  extra :
    attrs : [bool use_mkldnn = false]

2838
- op : trunc
2839
  inputs :
2840
    input : X
2841 2842
  outputs :
    out : Out
2843

2844 2845 2846 2847
- op : truncated_gaussian_random
  outputs :
    out : Out

2848 2849 2850 2851 2852 2853
- op : unbind
  inputs :
    input : X
  outputs :
    out : Out

2854 2855 2856 2857 2858 2859
- op : unfold
  inputs :
    x : X
  outputs :
    out : Y

2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876
- 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]

2877 2878 2879 2880 2881 2882 2883
- op : uniform_inplace (uniform_random_inplace)
  backward : uniform_inplace_grad(uniform_random_inplace_grad)
  inputs :
    x : X
  outputs :
    out : Out

L
lzydev 已提交
2884 2885 2886 2887 2888 2889 2890 2891 2892
- op : unique
  inputs :
    {x : X}
  outputs :
    {out : Out, indices : Indices, inverse : Index, counts : Counts}
  get_expected_kernel_type :
    unique : GetUniqueExpectedKernelType
  manual_signature : [unique]

2893 2894 2895 2896 2897 2898
- op : unique_consecutive
  inputs :
    x : X
  outputs :
    {out : Out, index : Index, counts : Counts}

2899 2900 2901 2902 2903 2904 2905 2906 2907 2908
- op : unpool
  inputs :
    {x : X, indices: Indices}
  outputs :
    out : Out
  int_array :
    output_size:
      data_type : int
      support_tensor : true

2909 2910 2911 2912 2913 2914
- op : unpool3d
  inputs :
    {x : X, indices: Indices}
  outputs :
    out : Out

2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930
- 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]

2931 2932 2933 2934 2935 2936 2937
- op : unstack
  backward : unstack_grad
  inputs :
    x : X
  outputs :
    out : Y

2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949
- 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

2950 2951 2952 2953 2954 2955
- op : viterbi_decode
  inputs :
    {potentials : Input, transition_params : Transition, lengths : Length}
  outputs :
    {scores : Scores, path : Path}

2956 2957 2958 2959 2960 2961 2962
- op : warpctc
  backward : warpctc_grad
  inputs :
    {logits : Logits, label : Label, logits_length : LogitsLength, labels_length : LabelLength}
  outputs :
    {warpctcgrad : WarpCTCGrad, loss : Loss}

2963 2964 2965 2966 2967 2968 2969
- op : where
  backward : where_grad
  inputs :
    {condition : Condition, x : X, y : Y}
  outputs :
    out : Out

2970 2971
- op : while
  backward : while_grad
2972
  extra :
2973
    attrs : ['str[] skip_eager_deletion_vars = {}']
2974

2975 2976 2977 2978 2979 2980
- op : yolo_box
  inputs :
    {x : X, img_size : ImgSize}
  outputs :
    {boxes : Boxes, scores : Scores}

2981 2982 2983 2984 2985 2986 2987 2988 2989 2990
- op : yolo_loss (yolov3_loss)
  backward: yolo_loss_grad (yolov3_loss_grad)
  inputs :
    {x : X,  gt_box : GTBox, gt_label : GTLabel ,gt_score : GTScore}
  outputs :
    {loss : Loss , objectness_mask : ObjectnessMask, gt_match_mask : GTMatchMask}
  get_expected_kernel_type :
    yolo_loss : GetYoloLossExpectedKernelType
    yolo_loss_grad : GetYoloLossExpectedKernelType

2991 2992 2993 2994 2995 2996 2997
- op: channel_shuffle
  inputs:
    {x: X}
  outputs:
    {out: Out}

- op: fetch (fetch_v2)
2998 2999 3000
  inputs: {x: X}
  outputs: {out: Out}

3001
- op: full_batch_size_like (fill_constant_batch_size_like)
3002 3003 3004 3005 3006 3007 3008 3009 3010 3011
  inputs:
    {input: Input}
  outputs:
    {out: Out}

- op: logspace
  inputs:
    {start: Start, stop: Stop, num: Num, base: Base}
  outputs:
    {out: Out}
3012

3013 3014 3015 3016 3017 3018 3019 3020 3021
- op: lu
  backward: lu_grad
  inputs:
    x: X
  outputs:
    {out: Out, pivots : Pivots, infos : Infos}
  attrs:
    pivot : pivots

Z
zhangyuqin1998 已提交
3022 3023 3024 3025 3026 3027
- op: reindex_graph (graph_reindex)
  inputs :
    {x : X, neighbors : Neighbors, count : Count, hashtable_value : HashTable_Value, hashtable_index : HashTable_Index}
  outputs :
    {reindex_src : Reindex_Src, reindex_dst : Reindex_Dst, out_nodes : Out_Nodes}

3028 3029 3030 3031 3032 3033
- op: rrelu
  inputs:
    {x: X}
  outputs:
    {out: Out, noise: Noise}

3034 3035 3036
- op: sigmoid_cross_entropy_with_logits
  backward: sigmoid_cross_entropy_with_logits_grad
  inputs :
3037
    {x: X, label: Label}
3038 3039
  outputs :
    out : Out
3040 3041 3042 3043 3044 3045 3046

- op: squared_l2_norm
  backward: squared_l2_norm_grad
  inputs :
    x : X
  outputs :
    out : Out
3047 3048 3049 3050 3051 3052 3053

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