1 /** 2 * Copyright 2023 Huawei Technologies Co., Ltd 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #ifndef MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_ 17 #define MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_ 18 19 #include <vector> 20 #include "ops/base_operator.h" 21 #include "ops/auto_generate/gen_ops_name.h" 22 23 namespace mindspore::ops { 24 class MIND_API ACosGrad : public BaseOperator { 25 public: 26 MIND_API_BASE_MEMBER(ACosGrad); ACosGrad()27 ACosGrad() : BaseOperator(kNameACosGrad) {} 28 }; 29 30 class MIND_API AbsGrad : public BaseOperator { 31 public: 32 MIND_API_BASE_MEMBER(AbsGrad); AbsGrad()33 AbsGrad() : BaseOperator(kNameAbsGrad) {} 34 }; 35 36 class MIND_API Abs : public BaseOperator { 37 public: 38 MIND_API_BASE_MEMBER(Abs); Abs()39 Abs() : BaseOperator(kNameAbs) {} 40 }; 41 42 class MIND_API ACos : public BaseOperator { 43 public: 44 MIND_API_BASE_MEMBER(ACos); ACos()45 ACos() : BaseOperator(kNameACos) {} 46 }; 47 48 class MIND_API AcoshGrad : public BaseOperator { 49 public: 50 MIND_API_BASE_MEMBER(AcoshGrad); AcoshGrad()51 AcoshGrad() : BaseOperator(kNameAcoshGrad) {} 52 }; 53 54 class MIND_API Acosh : public BaseOperator { 55 public: 56 MIND_API_BASE_MEMBER(Acosh); Acosh()57 Acosh() : BaseOperator(kNameAcosh) {} 58 }; 59 60 class MIND_API AdamWeightDecay : public BaseOperator { 61 public: 62 MIND_API_BASE_MEMBER(AdamWeightDecay); AdamWeightDecay()63 AdamWeightDecay() : BaseOperator(kNameAdamWeightDecay) {} 64 void set_use_locking(const bool &use_locking); 65 bool get_use_locking() const; 66 }; 67 68 class MIND_API AdamW : public BaseOperator { 69 public: 70 MIND_API_BASE_MEMBER(AdamW); AdamW()71 AdamW() : BaseOperator(kNameAdamW) {} 72 }; 73 74 class MIND_API AddExt : public BaseOperator { 75 public: 76 MIND_API_BASE_MEMBER(AddExt); AddExt()77 AddExt() : BaseOperator(kNameAddExt) {} 78 }; 79 80 class MIND_API AddLayerNormV2 : public BaseOperator { 81 public: 82 MIND_API_BASE_MEMBER(AddLayerNormV2); AddLayerNormV2()83 AddLayerNormV2() : BaseOperator(kNameAddLayerNormV2) {} 84 }; 85 86 class MIND_API Add : public BaseOperator { 87 public: 88 MIND_API_BASE_MEMBER(Add); Add()89 Add() : BaseOperator(kNameAdd) {} 90 }; 91 92 class MIND_API Addcdiv : public BaseOperator { 93 public: 94 MIND_API_BASE_MEMBER(Addcdiv); Addcdiv()95 Addcdiv() : BaseOperator(kNameAddcdiv) {} 96 }; 97 98 class MIND_API Addcmul : public BaseOperator { 99 public: 100 MIND_API_BASE_MEMBER(Addcmul); Addcmul()101 Addcmul() : BaseOperator(kNameAddcmul) {} 102 }; 103 104 class MIND_API Addmm : public BaseOperator { 105 public: 106 MIND_API_BASE_MEMBER(Addmm); Addmm()107 Addmm() : BaseOperator(kNameAddmm) {} 108 }; 109 110 class MIND_API AddN : public BaseOperator { 111 public: 112 MIND_API_BASE_MEMBER(AddN); AddN()113 AddN() : BaseOperator(kNameAddN) {} 114 }; 115 116 class MIND_API Angle : public BaseOperator { 117 public: 118 MIND_API_BASE_MEMBER(Angle); Angle()119 Angle() : BaseOperator(kNameAngle) {} 120 }; 121 122 class MIND_API ApplyCamePart1 : public BaseOperator { 123 public: 124 MIND_API_BASE_MEMBER(ApplyCamePart1); ApplyCamePart1()125 ApplyCamePart1() : BaseOperator(kNameApplyCamePart1) {} 126 }; 127 128 class MIND_API ApplyCamePart2 : public BaseOperator { 129 public: 130 MIND_API_BASE_MEMBER(ApplyCamePart2); ApplyCamePart2()131 ApplyCamePart2() : BaseOperator(kNameApplyCamePart2) {} 132 }; 133 134 class MIND_API ApplyCamePart3 : public BaseOperator { 135 public: 136 MIND_API_BASE_MEMBER(ApplyCamePart3); ApplyCamePart3()137 ApplyCamePart3() : BaseOperator(kNameApplyCamePart3) {} 138 }; 139 140 class MIND_API ApplyCamePart4 : public BaseOperator { 141 public: 142 MIND_API_BASE_MEMBER(ApplyCamePart4); ApplyCamePart4()143 ApplyCamePart4() : BaseOperator(kNameApplyCamePart4) {} 144 }; 145 146 class MIND_API ApplyRotaryPosEmb : public BaseOperator { 147 public: 148 MIND_API_BASE_MEMBER(ApplyRotaryPosEmb); ApplyRotaryPosEmb()149 ApplyRotaryPosEmb() : BaseOperator(kNameApplyRotaryPosEmb) {} 150 void set_cos_format(const int64_t &cos_format); 151 int64_t get_cos_format() const; 152 }; 153 154 class MIND_API Arange : public BaseOperator { 155 public: 156 MIND_API_BASE_MEMBER(Arange); Arange()157 Arange() : BaseOperator(kNameArange) {} 158 }; 159 160 class MIND_API ArgMaxExt : public BaseOperator { 161 public: 162 MIND_API_BASE_MEMBER(ArgMaxExt); ArgMaxExt()163 ArgMaxExt() : BaseOperator(kNameArgMaxExt) {} 164 }; 165 166 class MIND_API Argmax : public BaseOperator { 167 public: 168 MIND_API_BASE_MEMBER(Argmax); Argmax()169 Argmax() : BaseOperator(kNameArgmax) {} 170 void set_axis(const int64_t &axis); 171 int64_t get_axis() const; 172 void set_output_type(const int64_t &output_type); 173 int64_t get_output_type() const; 174 }; 175 176 class MIND_API ArgMaxWithValue : public BaseOperator { 177 public: 178 MIND_API_BASE_MEMBER(ArgMaxWithValue); ArgMaxWithValue()179 ArgMaxWithValue() : BaseOperator(kNameArgMaxWithValue) {} 180 void set_axis(const int64_t &axis); 181 int64_t get_axis() const; 182 void set_keep_dims(const bool &keep_dims); 183 bool get_keep_dims() const; 184 }; 185 186 class MIND_API Argmin : public BaseOperator { 187 public: 188 MIND_API_BASE_MEMBER(Argmin); Argmin()189 Argmin() : BaseOperator(kNameArgmin) {} 190 void set_axis(const int64_t &axis); 191 int64_t get_axis() const; 192 void set_output_type(const int64_t &output_type); 193 int64_t get_output_type() const; 194 }; 195 196 class MIND_API ArgMinWithValue : public BaseOperator { 197 public: 198 MIND_API_BASE_MEMBER(ArgMinWithValue); ArgMinWithValue()199 ArgMinWithValue() : BaseOperator(kNameArgMinWithValue) {} 200 void set_axis(const int64_t &axis); 201 int64_t get_axis() const; 202 void set_keep_dims(const bool &keep_dims); 203 bool get_keep_dims() const; 204 }; 205 206 class MIND_API AsinGrad : public BaseOperator { 207 public: 208 MIND_API_BASE_MEMBER(AsinGrad); AsinGrad()209 AsinGrad() : BaseOperator(kNameAsinGrad) {} 210 }; 211 212 class MIND_API Asin : public BaseOperator { 213 public: 214 MIND_API_BASE_MEMBER(Asin); Asin()215 Asin() : BaseOperator(kNameAsin) {} 216 }; 217 218 class MIND_API AsinhGrad : public BaseOperator { 219 public: 220 MIND_API_BASE_MEMBER(AsinhGrad); AsinhGrad()221 AsinhGrad() : BaseOperator(kNameAsinhGrad) {} 222 }; 223 224 class MIND_API Asinh : public BaseOperator { 225 public: 226 MIND_API_BASE_MEMBER(Asinh); Asinh()227 Asinh() : BaseOperator(kNameAsinh) {} 228 }; 229 230 class MIND_API AssignAdd : public BaseOperator { 231 public: 232 MIND_API_BASE_MEMBER(AssignAdd); AssignAdd()233 AssignAdd() : BaseOperator(kNameAssignAdd) {} 234 }; 235 236 class MIND_API Assign : public BaseOperator { 237 public: 238 MIND_API_BASE_MEMBER(Assign); Assign()239 Assign() : BaseOperator(kNameAssign) {} 240 }; 241 242 class MIND_API Atan2Ext : public BaseOperator { 243 public: 244 MIND_API_BASE_MEMBER(Atan2Ext); Atan2Ext()245 Atan2Ext() : BaseOperator(kNameAtan2Ext) {} 246 }; 247 248 class MIND_API Atan2 : public BaseOperator { 249 public: 250 MIND_API_BASE_MEMBER(Atan2); Atan2()251 Atan2() : BaseOperator(kNameAtan2) {} 252 }; 253 254 class MIND_API AtanGrad : public BaseOperator { 255 public: 256 MIND_API_BASE_MEMBER(AtanGrad); AtanGrad()257 AtanGrad() : BaseOperator(kNameAtanGrad) {} 258 }; 259 260 class MIND_API Atan : public BaseOperator { 261 public: 262 MIND_API_BASE_MEMBER(Atan); Atan()263 Atan() : BaseOperator(kNameAtan) {} 264 }; 265 266 class MIND_API Atanh : public BaseOperator { 267 public: 268 MIND_API_BASE_MEMBER(Atanh); Atanh()269 Atanh() : BaseOperator(kNameAtanh) {} 270 }; 271 272 class MIND_API AvgPool2DGrad : public BaseOperator { 273 public: 274 MIND_API_BASE_MEMBER(AvgPool2DGrad); AvgPool2DGrad()275 AvgPool2DGrad() : BaseOperator(kNameAvgPool2DGrad) {} 276 }; 277 278 class MIND_API AvgPool2D : public BaseOperator { 279 public: 280 MIND_API_BASE_MEMBER(AvgPool2D); AvgPool2D()281 AvgPool2D() : BaseOperator(kNameAvgPool2D) {} 282 }; 283 284 class MIND_API AvgPoolGrad : public BaseOperator { 285 public: 286 MIND_API_BASE_MEMBER(AvgPoolGrad); AvgPoolGrad()287 AvgPoolGrad() : BaseOperator(kNameAvgPoolGrad) {} 288 void set_kernel_size(const std::vector<int64_t> &kernel_size); 289 std::vector<int64_t> get_kernel_size() const; 290 void set_strides(const std::vector<int64_t> &strides); 291 std::vector<int64_t> get_strides() const; 292 void set_pad_mode(const int64_t &pad_mode); 293 int64_t get_pad_mode() const; 294 void set_data_format(const int64_t &data_format); 295 int64_t get_data_format() const; 296 }; 297 298 class MIND_API AvgPool : public BaseOperator { 299 public: 300 MIND_API_BASE_MEMBER(AvgPool); AvgPool()301 AvgPool() : BaseOperator(kNameAvgPool) {} 302 void set_kernel_size(const std::vector<int64_t> &kernel_size); 303 std::vector<int64_t> get_kernel_size() const; 304 void set_strides(const std::vector<int64_t> &strides); 305 std::vector<int64_t> get_strides() const; 306 void set_pad_mode(const int64_t &pad_mode); 307 int64_t get_pad_mode() const; 308 void set_data_format(const int64_t &data_format); 309 int64_t get_data_format() const; 310 }; 311 312 class MIND_API BatchMatMul : public BaseOperator { 313 public: 314 MIND_API_BASE_MEMBER(BatchMatMul); BatchMatMul()315 BatchMatMul() : BaseOperator(kNameBatchMatMul) {} 316 void set_transpose_a(const bool &transpose_a); 317 bool get_transpose_a() const; 318 void set_transpose_b(const bool &transpose_b); 319 bool get_transpose_b() const; 320 }; 321 322 class MIND_API BatchNormExt : public BaseOperator { 323 public: 324 MIND_API_BASE_MEMBER(BatchNormExt); BatchNormExt()325 BatchNormExt() : BaseOperator(kNameBatchNormExt) {} 326 }; 327 328 class MIND_API BatchNormGradExt : public BaseOperator { 329 public: 330 MIND_API_BASE_MEMBER(BatchNormGradExt); BatchNormGradExt()331 BatchNormGradExt() : BaseOperator(kNameBatchNormGradExt) {} 332 void set_training(const bool &training); 333 bool get_training() const; 334 void set_eps(const float &eps); 335 float get_eps() const; 336 }; 337 338 class MIND_API BatchNormGradGrad : public BaseOperator { 339 public: 340 MIND_API_BASE_MEMBER(BatchNormGradGrad); BatchNormGradGrad()341 BatchNormGradGrad() : BaseOperator(kNameBatchNormGradGrad) {} 342 void set_is_training(const bool &is_training); 343 bool get_is_training() const; 344 void set_epsilon(const float &epsilon); 345 float get_epsilon() const; 346 void set_data_format(const int64_t &data_format); 347 int64_t get_data_format() const; 348 }; 349 350 class MIND_API BatchNormGrad : public BaseOperator { 351 public: 352 MIND_API_BASE_MEMBER(BatchNormGrad); BatchNormGrad()353 BatchNormGrad() : BaseOperator(kNameBatchNormGrad) {} 354 void set_is_training(const bool &is_training); 355 bool get_is_training() const; 356 void set_epsilon(const float &epsilon); 357 float get_epsilon() const; 358 void set_data_format(const int64_t &data_format); 359 int64_t get_data_format() const; 360 }; 361 362 class MIND_API BatchNormGradWithActivation : public BaseOperator { 363 public: 364 MIND_API_BASE_MEMBER(BatchNormGradWithActivation); BatchNormGradWithActivation()365 BatchNormGradWithActivation() : BaseOperator(kNameBatchNormGradWithActivation) {} 366 void set_is_training(const bool &is_training); 367 bool get_is_training() const; 368 void set_epsilon(const float &epsilon); 369 float get_epsilon() const; 370 void set_data_format(const int64_t &data_format); 371 int64_t get_data_format() const; 372 }; 373 374 class MIND_API BatchNorm : public BaseOperator { 375 public: 376 MIND_API_BASE_MEMBER(BatchNorm); BatchNorm()377 BatchNorm() : BaseOperator(kNameBatchNorm) {} 378 void set_is_training(const bool &is_training); 379 bool get_is_training() const; 380 void set_epsilon(const float &epsilon); 381 float get_epsilon() const; 382 void set_momentum(const float &momentum); 383 float get_momentum() const; 384 void set_data_format(const int64_t &data_format); 385 int64_t get_data_format() const; 386 }; 387 388 class MIND_API BatchNormWithActivation : public BaseOperator { 389 public: 390 MIND_API_BASE_MEMBER(BatchNormWithActivation); BatchNormWithActivation()391 BatchNormWithActivation() : BaseOperator(kNameBatchNormWithActivation) {} 392 void set_is_training(const bool &is_training); 393 bool get_is_training() const; 394 void set_epsilon(const float &epsilon); 395 float get_epsilon() const; 396 void set_momentum(const float &momentum); 397 float get_momentum() const; 398 void set_data_format(const int64_t &data_format); 399 int64_t get_data_format() const; 400 }; 401 402 class MIND_API BatchNormWithAddAndActivation : public BaseOperator { 403 public: 404 MIND_API_BASE_MEMBER(BatchNormWithAddAndActivation); BatchNormWithAddAndActivation()405 BatchNormWithAddAndActivation() : BaseOperator(kNameBatchNormWithAddAndActivation) {} 406 void set_is_training(const bool &is_training); 407 bool get_is_training() const; 408 void set_epsilon(const float &epsilon); 409 float get_epsilon() const; 410 void set_momentum(const float &momentum); 411 float get_momentum() const; 412 void set_data_format(const int64_t &data_format); 413 int64_t get_data_format() const; 414 }; 415 416 class MIND_API Betainc : public BaseOperator { 417 public: 418 MIND_API_BASE_MEMBER(Betainc); Betainc()419 Betainc() : BaseOperator(kNameBetainc) {} 420 }; 421 422 class MIND_API BiasAddGrad : public BaseOperator { 423 public: 424 MIND_API_BASE_MEMBER(BiasAddGrad); BiasAddGrad()425 BiasAddGrad() : BaseOperator(kNameBiasAddGrad) {} 426 void set_data_format(const int64_t &data_format); 427 int64_t get_data_format() const; 428 }; 429 430 class MIND_API BiasAdd : public BaseOperator { 431 public: 432 MIND_API_BASE_MEMBER(BiasAdd); BiasAdd()433 BiasAdd() : BaseOperator(kNameBiasAdd) {} 434 void set_data_format(const int64_t &data_format); 435 int64_t get_data_format() const; 436 }; 437 438 class MIND_API BinaryCrossEntropyGrad : public BaseOperator { 439 public: 440 MIND_API_BASE_MEMBER(BinaryCrossEntropyGrad); BinaryCrossEntropyGrad()441 BinaryCrossEntropyGrad() : BaseOperator(kNameBinaryCrossEntropyGrad) {} 442 void set_reduction(const int64_t &reduction); 443 int64_t get_reduction() const; 444 }; 445 446 class MIND_API BinaryCrossEntropy : public BaseOperator { 447 public: 448 MIND_API_BASE_MEMBER(BinaryCrossEntropy); BinaryCrossEntropy()449 BinaryCrossEntropy() : BaseOperator(kNameBinaryCrossEntropy) {} 450 void set_reduction(const int64_t &reduction); 451 int64_t get_reduction() const; 452 }; 453 454 class MIND_API BinaryCrossEntropyWithLogitsBackward : public BaseOperator { 455 public: 456 MIND_API_BASE_MEMBER(BinaryCrossEntropyWithLogitsBackward); BinaryCrossEntropyWithLogitsBackward()457 BinaryCrossEntropyWithLogitsBackward() : BaseOperator(kNameBinaryCrossEntropyWithLogitsBackward) {} 458 }; 459 460 class MIND_API BCEWithLogitsLoss : public BaseOperator { 461 public: 462 MIND_API_BASE_MEMBER(BCEWithLogitsLoss); BCEWithLogitsLoss()463 BCEWithLogitsLoss() : BaseOperator(kNameBCEWithLogitsLoss) {} 464 void set_reduction(const int64_t &reduction); 465 int64_t get_reduction() const; 466 }; 467 468 class MIND_API BatchMatMulExt : public BaseOperator { 469 public: 470 MIND_API_BASE_MEMBER(BatchMatMulExt); BatchMatMulExt()471 BatchMatMulExt() : BaseOperator(kNameBatchMatMulExt) {} 472 }; 473 474 class MIND_API BoolNot : public BaseOperator { 475 public: 476 MIND_API_BASE_MEMBER(BoolNot); BoolNot()477 BoolNot() : BaseOperator(kNameBoolNot) {} 478 }; 479 480 class MIND_API BroadcastTo : public BaseOperator { 481 public: 482 MIND_API_BASE_MEMBER(BroadcastTo); BroadcastTo()483 BroadcastTo() : BaseOperator(kNameBroadcastTo) {} 484 void set_shape(const std::vector<int64_t> &shape); 485 std::vector<int64_t> get_shape() const; 486 }; 487 488 class MIND_API Cast : public BaseOperator { 489 public: 490 MIND_API_BASE_MEMBER(Cast); Cast()491 Cast() : BaseOperator(kNameCast) {} 492 }; 493 494 class MIND_API Ceil : public BaseOperator { 495 public: 496 MIND_API_BASE_MEMBER(Ceil); Ceil()497 Ceil() : BaseOperator(kNameCeil) {} 498 }; 499 500 class MIND_API CeLU : public BaseOperator { 501 public: 502 MIND_API_BASE_MEMBER(CeLU); CeLU()503 CeLU() : BaseOperator(kNameCeLU) {} 504 void set_alpha(const float &alpha); 505 float get_alpha() const; 506 }; 507 508 class MIND_API CholeskyGrad : public BaseOperator { 509 public: 510 MIND_API_BASE_MEMBER(CholeskyGrad); CholeskyGrad()511 CholeskyGrad() : BaseOperator(kNameCholeskyGrad) {} 512 }; 513 514 class MIND_API CholeskyInverse : public BaseOperator { 515 public: 516 MIND_API_BASE_MEMBER(CholeskyInverse); CholeskyInverse()517 CholeskyInverse() : BaseOperator(kNameCholeskyInverse) {} 518 void set_upper(const bool &upper); 519 bool get_upper() const; 520 }; 521 522 class MIND_API Cholesky : public BaseOperator { 523 public: 524 MIND_API_BASE_MEMBER(Cholesky); Cholesky()525 Cholesky() : BaseOperator(kNameCholesky) {} 526 void set_upper(const bool &upper); 527 bool get_upper() const; 528 }; 529 530 class MIND_API Chunk : public BaseOperator { 531 public: 532 MIND_API_BASE_MEMBER(Chunk); Chunk()533 Chunk() : BaseOperator(kNameChunk) {} 534 }; 535 536 class MIND_API ClampScalar : public BaseOperator { 537 public: 538 MIND_API_BASE_MEMBER(ClampScalar); ClampScalar()539 ClampScalar() : BaseOperator(kNameClampScalar) {} 540 }; 541 542 class MIND_API ClampTensor : public BaseOperator { 543 public: 544 MIND_API_BASE_MEMBER(ClampTensor); ClampTensor()545 ClampTensor() : BaseOperator(kNameClampTensor) {} 546 }; 547 548 class MIND_API Col2ImExt : public BaseOperator { 549 public: 550 MIND_API_BASE_MEMBER(Col2ImExt); Col2ImExt()551 Col2ImExt() : BaseOperator(kNameCol2ImExt) {} 552 }; 553 554 class MIND_API Col2ImGrad : public BaseOperator { 555 public: 556 MIND_API_BASE_MEMBER(Col2ImGrad); Col2ImGrad()557 Col2ImGrad() : BaseOperator(kNameCol2ImGrad) {} 558 }; 559 560 class MIND_API Complex : public BaseOperator { 561 public: 562 MIND_API_BASE_MEMBER(Complex); Complex()563 Complex() : BaseOperator(kNameComplex) {} 564 }; 565 566 class MIND_API Concat : public BaseOperator { 567 public: 568 MIND_API_BASE_MEMBER(Concat); Concat()569 Concat() : BaseOperator(kNameConcat) {} 570 void set_axis(const int64_t &axis); 571 int64_t get_axis() const; 572 }; 573 574 class MIND_API Conj : public BaseOperator { 575 public: 576 MIND_API_BASE_MEMBER(Conj); Conj()577 Conj() : BaseOperator(kNameConj) {} 578 }; 579 580 class MIND_API ConstantPadND : public BaseOperator { 581 public: 582 MIND_API_BASE_MEMBER(ConstantPadND); ConstantPadND()583 ConstantPadND() : BaseOperator(kNameConstantPadND) {} 584 }; 585 586 class MIND_API Contiguous : public BaseOperator { 587 public: 588 MIND_API_BASE_MEMBER(Contiguous); Contiguous()589 Contiguous() : BaseOperator(kNameContiguous) {} 590 }; 591 592 class MIND_API ConvolutionGrad : public BaseOperator { 593 public: 594 MIND_API_BASE_MEMBER(ConvolutionGrad); ConvolutionGrad()595 ConvolutionGrad() : BaseOperator(kNameConvolutionGrad) {} 596 void set_stride(const std::vector<int64_t> &stride); 597 std::vector<int64_t> get_stride() const; 598 void set_padding(const std::vector<int64_t> &padding); 599 std::vector<int64_t> get_padding() const; 600 void set_dilation(const std::vector<int64_t> &dilation); 601 std::vector<int64_t> get_dilation() const; 602 void set_transposed(const bool &transposed); 603 bool get_transposed() const; 604 void set_output_padding(const std::vector<int64_t> &output_padding); 605 std::vector<int64_t> get_output_padding() const; 606 void set_groups(const int64_t &groups); 607 int64_t get_groups() const; 608 void set_output_mask(const std::vector<int64_t> &output_mask); 609 std::vector<int64_t> get_output_mask() const; 610 }; 611 612 class MIND_API Convolution : public BaseOperator { 613 public: 614 MIND_API_BASE_MEMBER(Convolution); Convolution()615 Convolution() : BaseOperator(kNameConvolution) {} 616 void set_stride(const std::vector<int64_t> &stride); 617 std::vector<int64_t> get_stride() const; 618 void set_padding(const std::vector<int64_t> &padding); 619 std::vector<int64_t> get_padding() const; 620 void set_dilation(const std::vector<int64_t> &dilation); 621 std::vector<int64_t> get_dilation() const; 622 void set_transposed(const bool &transposed); 623 bool get_transposed() const; 624 void set_output_padding(const std::vector<int64_t> &output_padding); 625 std::vector<int64_t> get_output_padding() const; 626 void set_groups(const int64_t &groups); 627 int64_t get_groups() const; 628 }; 629 630 class MIND_API Copy : public BaseOperator { 631 public: 632 MIND_API_BASE_MEMBER(Copy); Copy()633 Copy() : BaseOperator(kNameCopy) {} 634 }; 635 636 class MIND_API Correlate : public BaseOperator { 637 public: 638 MIND_API_BASE_MEMBER(Correlate); Correlate()639 Correlate() : BaseOperator(kNameCorrelate) {} 640 void set_mode(const int64_t &mode); 641 int64_t get_mode() const; 642 }; 643 644 class MIND_API Cos : public BaseOperator { 645 public: 646 MIND_API_BASE_MEMBER(Cos); Cos()647 Cos() : BaseOperator(kNameCos) {} 648 }; 649 650 class MIND_API Cosh : public BaseOperator { 651 public: 652 MIND_API_BASE_MEMBER(Cosh); Cosh()653 Cosh() : BaseOperator(kNameCosh) {} 654 }; 655 656 class MIND_API CumProd : public BaseOperator { 657 public: 658 MIND_API_BASE_MEMBER(CumProd); CumProd()659 CumProd() : BaseOperator(kNameCumProd) {} 660 void set_exclusive(const bool &exclusive); 661 bool get_exclusive() const; 662 void set_reverse(const bool &reverse); 663 bool get_reverse() const; 664 }; 665 666 class MIND_API CumSum : public BaseOperator { 667 public: 668 MIND_API_BASE_MEMBER(CumSum); CumSum()669 CumSum() : BaseOperator(kNameCumSum) {} 670 void set_exclusive(const bool &exclusive); 671 bool get_exclusive() const; 672 void set_reverse(const bool &reverse); 673 bool get_reverse() const; 674 }; 675 676 class MIND_API Cummax : public BaseOperator { 677 public: 678 MIND_API_BASE_MEMBER(Cummax); Cummax()679 Cummax() : BaseOperator(kNameCummax) {} 680 void set_axis(const int64_t &axis); 681 int64_t get_axis() const; 682 }; 683 684 class MIND_API Cummin : public BaseOperator { 685 public: 686 MIND_API_BASE_MEMBER(Cummin); Cummin()687 Cummin() : BaseOperator(kNameCummin) {} 688 void set_axis(const int64_t &axis); 689 int64_t get_axis() const; 690 }; 691 692 class MIND_API CumsumExt : public BaseOperator { 693 public: 694 MIND_API_BASE_MEMBER(CumsumExt); CumsumExt()695 CumsumExt() : BaseOperator(kNameCumsumExt) {} 696 }; 697 698 class MIND_API DCT : public BaseOperator { 699 public: 700 MIND_API_BASE_MEMBER(DCT); DCT()701 DCT() : BaseOperator(kNameDCT) {} 702 }; 703 704 class MIND_API DecoderKVCache : public BaseOperator { 705 public: 706 MIND_API_BASE_MEMBER(DecoderKVCache); DecoderKVCache()707 DecoderKVCache() : BaseOperator(kNameDecoderKVCache) {} 708 }; 709 710 class MIND_API Dense : public BaseOperator { 711 public: 712 MIND_API_BASE_MEMBER(Dense); Dense()713 Dense() : BaseOperator(kNameDense) {} 714 }; 715 716 class MIND_API Diag : public BaseOperator { 717 public: 718 MIND_API_BASE_MEMBER(Diag); Diag()719 Diag() : BaseOperator(kNameDiag) {} 720 }; 721 722 class MIND_API Diagonal : public BaseOperator { 723 public: 724 MIND_API_BASE_MEMBER(Diagonal); Diagonal()725 Diagonal() : BaseOperator(kNameDiagonal) {} 726 void set_offset(const int64_t &offset); 727 int64_t get_offset() const; 728 void set_dim1(const int64_t &dim1); 729 int64_t get_dim1() const; 730 void set_dim2(const int64_t &dim2); 731 int64_t get_dim2() const; 732 }; 733 734 class MIND_API Div : public BaseOperator { 735 public: 736 MIND_API_BASE_MEMBER(Div); Div()737 Div() : BaseOperator(kNameDiv) {} 738 }; 739 740 class MIND_API DivMod : public BaseOperator { 741 public: 742 MIND_API_BASE_MEMBER(DivMod); DivMod()743 DivMod() : BaseOperator(kNameDivMod) {} 744 }; 745 746 class MIND_API Dot : public BaseOperator { 747 public: 748 MIND_API_BASE_MEMBER(Dot); Dot()749 Dot() : BaseOperator(kNameDot) {} 750 }; 751 752 class MIND_API DropoutDoMaskExt : public BaseOperator { 753 public: 754 MIND_API_BASE_MEMBER(DropoutDoMaskExt); DropoutDoMaskExt()755 DropoutDoMaskExt() : BaseOperator(kNameDropoutDoMaskExt) {} 756 }; 757 758 class MIND_API DropoutExt : public BaseOperator { 759 public: 760 MIND_API_BASE_MEMBER(DropoutExt); DropoutExt()761 DropoutExt() : BaseOperator(kNameDropoutExt) {} 762 }; 763 764 class MIND_API DropoutGenMaskExt : public BaseOperator { 765 public: 766 MIND_API_BASE_MEMBER(DropoutGenMaskExt); DropoutGenMaskExt()767 DropoutGenMaskExt() : BaseOperator(kNameDropoutGenMaskExt) {} 768 }; 769 770 class MIND_API DropoutGradExt : public BaseOperator { 771 public: 772 MIND_API_BASE_MEMBER(DropoutGradExt); DropoutGradExt()773 DropoutGradExt() : BaseOperator(kNameDropoutGradExt) {} 774 }; 775 776 class MIND_API Dropout : public BaseOperator { 777 public: 778 MIND_API_BASE_MEMBER(Dropout); Dropout()779 Dropout() : BaseOperator(kNameDropout) {} 780 void set_keep_prob(const float &keep_prob); 781 float get_keep_prob() const; 782 void set_Seed0(const int64_t &Seed0); 783 int64_t get_Seed0() const; 784 void set_Seed1(const int64_t &Seed1); 785 int64_t get_Seed1() const; 786 }; 787 788 class MIND_API Eig : public BaseOperator { 789 public: 790 MIND_API_BASE_MEMBER(Eig); Eig()791 Eig() : BaseOperator(kNameEig) {} 792 void set_compute_v(const bool &compute_v); 793 bool get_compute_v() const; 794 }; 795 796 class MIND_API EluExt : public BaseOperator { 797 public: 798 MIND_API_BASE_MEMBER(EluExt); EluExt()799 EluExt() : BaseOperator(kNameEluExt) {} 800 void set_alpha(const float &alpha); 801 float get_alpha() const; 802 }; 803 804 class MIND_API EluGradExt : public BaseOperator { 805 public: 806 MIND_API_BASE_MEMBER(EluGradExt); EluGradExt()807 EluGradExt() : BaseOperator(kNameEluGradExt) {} 808 }; 809 810 class MIND_API EluGrad : public BaseOperator { 811 public: 812 MIND_API_BASE_MEMBER(EluGrad); EluGrad()813 EluGrad() : BaseOperator(kNameEluGrad) {} 814 }; 815 816 class MIND_API Elu : public BaseOperator { 817 public: 818 MIND_API_BASE_MEMBER(Elu); Elu()819 Elu() : BaseOperator(kNameElu) {} 820 void set_alpha(const float &alpha); 821 float get_alpha() const; 822 }; 823 824 class MIND_API EmbeddingDenseBackward : public BaseOperator { 825 public: 826 MIND_API_BASE_MEMBER(EmbeddingDenseBackward); EmbeddingDenseBackward()827 EmbeddingDenseBackward() : BaseOperator(kNameEmbeddingDenseBackward) {} 828 }; 829 830 class MIND_API Embedding : public BaseOperator { 831 public: 832 MIND_API_BASE_MEMBER(Embedding); Embedding()833 Embedding() : BaseOperator(kNameEmbedding) {} 834 }; 835 836 class MIND_API Equal : public BaseOperator { 837 public: 838 MIND_API_BASE_MEMBER(Equal); Equal()839 Equal() : BaseOperator(kNameEqual) {} 840 }; 841 842 class MIND_API Erf : public BaseOperator { 843 public: 844 MIND_API_BASE_MEMBER(Erf); Erf()845 Erf() : BaseOperator(kNameErf) {} 846 }; 847 848 class MIND_API Erfc : public BaseOperator { 849 public: 850 MIND_API_BASE_MEMBER(Erfc); Erfc()851 Erfc() : BaseOperator(kNameErfc) {} 852 }; 853 854 class MIND_API Erfinv : public BaseOperator { 855 public: 856 MIND_API_BASE_MEMBER(Erfinv); Erfinv()857 Erfinv() : BaseOperator(kNameErfinv) {} 858 }; 859 860 class MIND_API Exp : public BaseOperator { 861 public: 862 MIND_API_BASE_MEMBER(Exp); Exp()863 Exp() : BaseOperator(kNameExp) {} 864 }; 865 866 class MIND_API ExpandDims : public BaseOperator { 867 public: 868 MIND_API_BASE_MEMBER(ExpandDims); ExpandDims()869 ExpandDims() : BaseOperator(kNameExpandDims) {} 870 }; 871 872 class MIND_API Expm1 : public BaseOperator { 873 public: 874 MIND_API_BASE_MEMBER(Expm1); Expm1()875 Expm1() : BaseOperator(kNameExpm1) {} 876 }; 877 878 class MIND_API ExtractImagePatches : public BaseOperator { 879 public: 880 MIND_API_BASE_MEMBER(ExtractImagePatches); ExtractImagePatches()881 ExtractImagePatches() : BaseOperator(kNameExtractImagePatches) {} 882 void set_ksizes(const std::vector<int64_t> &ksizes); 883 std::vector<int64_t> get_ksizes() const; 884 void set_strides(const std::vector<int64_t> &strides); 885 std::vector<int64_t> get_strides() const; 886 void set_rates(const std::vector<int64_t> &rates); 887 std::vector<int64_t> get_rates() const; 888 void set_padding(const int64_t &padding); 889 int64_t get_padding() const; 890 }; 891 892 class MIND_API Eye : public BaseOperator { 893 public: 894 MIND_API_BASE_MEMBER(Eye); Eye()895 Eye() : BaseOperator(kNameEye) {} 896 }; 897 898 class MIND_API FastGeLUGrad : public BaseOperator { 899 public: 900 MIND_API_BASE_MEMBER(FastGeLUGrad); FastGeLUGrad()901 FastGeLUGrad() : BaseOperator(kNameFastGeLUGrad) {} 902 }; 903 904 class MIND_API FastGeLU : public BaseOperator { 905 public: 906 MIND_API_BASE_MEMBER(FastGeLU); FastGeLU()907 FastGeLU() : BaseOperator(kNameFastGeLU) {} 908 }; 909 910 class MIND_API FFNExt : public BaseOperator { 911 public: 912 MIND_API_BASE_MEMBER(FFNExt); FFNExt()913 FFNExt() : BaseOperator(kNameFFNExt) {} 914 void set_activation(const int64_t &activation); 915 int64_t get_activation() const; 916 void set_inner_precise(const int64_t &inner_precise); 917 int64_t get_inner_precise() const; 918 }; 919 920 class MIND_API FFT2 : public BaseOperator { 921 public: 922 MIND_API_BASE_MEMBER(FFT2); FFT2()923 FFT2() : BaseOperator(kNameFFT2) {} 924 }; 925 926 class MIND_API FFT : public BaseOperator { 927 public: 928 MIND_API_BASE_MEMBER(FFT); FFT()929 FFT() : BaseOperator(kNameFFT) {} 930 }; 931 932 class MIND_API FFTShapeCopy : public BaseOperator { 933 public: 934 MIND_API_BASE_MEMBER(FFTShapeCopy); FFTShapeCopy()935 FFTShapeCopy() : BaseOperator(kNameFFTShapeCopy) {} 936 }; 937 938 class MIND_API FFTWithSize : public BaseOperator { 939 public: 940 MIND_API_BASE_MEMBER(FFTWithSize); FFTWithSize()941 FFTWithSize() : BaseOperator(kNameFFTWithSize) {} 942 void set_signal_ndim(const int64_t &signal_ndim); 943 int64_t get_signal_ndim() const; 944 void set_inverse(const bool &inverse); 945 bool get_inverse() const; 946 void set_real(const bool &real); 947 bool get_real() const; 948 void set_norm(const int64_t &norm); 949 int64_t get_norm() const; 950 void set_onesided(const bool &onesided); 951 bool get_onesided() const; 952 void set_signal_sizes(const std::vector<int64_t> &signal_sizes); 953 std::vector<int64_t> get_signal_sizes() const; 954 }; 955 956 class MIND_API FFTN : public BaseOperator { 957 public: 958 MIND_API_BASE_MEMBER(FFTN); FFTN()959 FFTN() : BaseOperator(kNameFFTN) {} 960 }; 961 962 class MIND_API FFTShift : public BaseOperator { 963 public: 964 MIND_API_BASE_MEMBER(FFTShift); FFTShift()965 FFTShift() : BaseOperator(kNameFFTShift) {} 966 }; 967 968 class MIND_API FillScalar : public BaseOperator { 969 public: 970 MIND_API_BASE_MEMBER(FillScalar); FillScalar()971 FillScalar() : BaseOperator(kNameFillScalar) {} 972 }; 973 974 class MIND_API FillTensor : public BaseOperator { 975 public: 976 MIND_API_BASE_MEMBER(FillTensor); FillTensor()977 FillTensor() : BaseOperator(kNameFillTensor) {} 978 }; 979 980 class MIND_API FlashAttentionScoreGrad : public BaseOperator { 981 public: 982 MIND_API_BASE_MEMBER(FlashAttentionScoreGrad); FlashAttentionScoreGrad()983 FlashAttentionScoreGrad() : BaseOperator(kNameFlashAttentionScoreGrad) {} 984 void set_head_num(const int64_t &head_num); 985 int64_t get_head_num() const; 986 void set_keep_prob(const float &keep_prob); 987 float get_keep_prob() const; 988 void set_scale_value(const float &scale_value); 989 float get_scale_value() const; 990 void set_pre_tokens(const int64_t &pre_tokens); 991 int64_t get_pre_tokens() const; 992 void set_next_tokens(const int64_t &next_tokens); 993 int64_t get_next_tokens() const; 994 void set_inner_precise(const int64_t &inner_precise); 995 int64_t get_inner_precise() const; 996 void set_input_layout(const int64_t &input_layout); 997 int64_t get_input_layout() const; 998 void set_sparse_mode(const int64_t &sparse_mode); 999 int64_t get_sparse_mode() const; 1000 }; 1001 1002 class MIND_API FlashAttentionScore : public BaseOperator { 1003 public: 1004 MIND_API_BASE_MEMBER(FlashAttentionScore); FlashAttentionScore()1005 FlashAttentionScore() : BaseOperator(kNameFlashAttentionScore) {} 1006 void set_head_num(const int64_t &head_num); 1007 int64_t get_head_num() const; 1008 void set_keep_prob(const float &keep_prob); 1009 float get_keep_prob() const; 1010 void set_scale_value(const float &scale_value); 1011 float get_scale_value() const; 1012 void set_pre_tokens(const int64_t &pre_tokens); 1013 int64_t get_pre_tokens() const; 1014 void set_next_tokens(const int64_t &next_tokens); 1015 int64_t get_next_tokens() const; 1016 void set_inner_precise(const int64_t &inner_precise); 1017 int64_t get_inner_precise() const; 1018 void set_input_layout(const int64_t &input_layout); 1019 int64_t get_input_layout() const; 1020 void set_sparse_mode(const int64_t &sparse_mode); 1021 int64_t get_sparse_mode() const; 1022 }; 1023 1024 class MIND_API FlattenExt : public BaseOperator { 1025 public: 1026 MIND_API_BASE_MEMBER(FlattenExt); FlattenExt()1027 FlattenExt() : BaseOperator(kNameFlattenExt) {} 1028 }; 1029 1030 class MIND_API Flatten : public BaseOperator { 1031 public: 1032 MIND_API_BASE_MEMBER(Flatten); Flatten()1033 Flatten() : BaseOperator(kNameFlatten) {} 1034 }; 1035 1036 class MIND_API FloorDiv : public BaseOperator { 1037 public: 1038 MIND_API_BASE_MEMBER(FloorDiv); FloorDiv()1039 FloorDiv() : BaseOperator(kNameFloorDiv) {} 1040 }; 1041 1042 class MIND_API FloorMod : public BaseOperator { 1043 public: 1044 MIND_API_BASE_MEMBER(FloorMod); FloorMod()1045 FloorMod() : BaseOperator(kNameFloorMod) {} 1046 }; 1047 1048 class MIND_API Floor : public BaseOperator { 1049 public: 1050 MIND_API_BASE_MEMBER(Floor); Floor()1051 Floor() : BaseOperator(kNameFloor) {} 1052 }; 1053 1054 class MIND_API GatherDGradV2 : public BaseOperator { 1055 public: 1056 MIND_API_BASE_MEMBER(GatherDGradV2); GatherDGradV2()1057 GatherDGradV2() : BaseOperator(kNameGatherDGradV2) {} 1058 }; 1059 1060 class MIND_API GatherD : public BaseOperator { 1061 public: 1062 MIND_API_BASE_MEMBER(GatherD); GatherD()1063 GatherD() : BaseOperator(kNameGatherD) {} 1064 }; 1065 1066 class MIND_API GatherNd : public BaseOperator { 1067 public: 1068 MIND_API_BASE_MEMBER(GatherNd); GatherNd()1069 GatherNd() : BaseOperator(kNameGatherNd) {} 1070 }; 1071 1072 class MIND_API Gather : public BaseOperator { 1073 public: 1074 MIND_API_BASE_MEMBER(Gather); Gather()1075 Gather() : BaseOperator(kNameGather) {} 1076 void set_batch_dims(const int64_t &batch_dims); 1077 int64_t get_batch_dims() const; 1078 }; 1079 1080 class MIND_API Gcd : public BaseOperator { 1081 public: 1082 MIND_API_BASE_MEMBER(Gcd); Gcd()1083 Gcd() : BaseOperator(kNameGcd) {} 1084 }; 1085 1086 class MIND_API GeLUGrad : public BaseOperator { 1087 public: 1088 MIND_API_BASE_MEMBER(GeLUGrad); GeLUGrad()1089 GeLUGrad() : BaseOperator(kNameGeLUGrad) {} 1090 }; 1091 1092 class MIND_API GeLU : public BaseOperator { 1093 public: 1094 MIND_API_BASE_MEMBER(GeLU); GeLU()1095 GeLU() : BaseOperator(kNameGeLU) {} 1096 }; 1097 1098 class MIND_API Generator : public BaseOperator { 1099 public: 1100 MIND_API_BASE_MEMBER(Generator); Generator()1101 Generator() : BaseOperator(kNameGenerator) {} 1102 }; 1103 1104 class MIND_API Geqrf : public BaseOperator { 1105 public: 1106 MIND_API_BASE_MEMBER(Geqrf); Geqrf()1107 Geqrf() : BaseOperator(kNameGeqrf) {} 1108 }; 1109 1110 class MIND_API GreaterEqual : public BaseOperator { 1111 public: 1112 MIND_API_BASE_MEMBER(GreaterEqual); GreaterEqual()1113 GreaterEqual() : BaseOperator(kNameGreaterEqual) {} 1114 }; 1115 1116 class MIND_API Greater : public BaseOperator { 1117 public: 1118 MIND_API_BASE_MEMBER(Greater); Greater()1119 Greater() : BaseOperator(kNameGreater) {} 1120 }; 1121 1122 class MIND_API GridSampler2DGrad : public BaseOperator { 1123 public: 1124 MIND_API_BASE_MEMBER(GridSampler2DGrad); GridSampler2DGrad()1125 GridSampler2DGrad() : BaseOperator(kNameGridSampler2DGrad) {} 1126 void set_interpolation_mode(const int64_t &interpolation_mode); 1127 int64_t get_interpolation_mode() const; 1128 void set_padding_mode(const int64_t &padding_mode); 1129 int64_t get_padding_mode() const; 1130 void set_align_corners(const bool &align_corners); 1131 bool get_align_corners() const; 1132 }; 1133 1134 class MIND_API GridSampler2D : public BaseOperator { 1135 public: 1136 MIND_API_BASE_MEMBER(GridSampler2D); GridSampler2D()1137 GridSampler2D() : BaseOperator(kNameGridSampler2D) {} 1138 void set_interpolation_mode(const int64_t &interpolation_mode); 1139 int64_t get_interpolation_mode() const; 1140 void set_padding_mode(const int64_t &padding_mode); 1141 int64_t get_padding_mode() const; 1142 void set_align_corners(const bool &align_corners); 1143 bool get_align_corners() const; 1144 }; 1145 1146 class MIND_API GridSampler3DGrad : public BaseOperator { 1147 public: 1148 MIND_API_BASE_MEMBER(GridSampler3DGrad); GridSampler3DGrad()1149 GridSampler3DGrad() : BaseOperator(kNameGridSampler3DGrad) {} 1150 void set_interpolation_mode(const int64_t &interpolation_mode); 1151 int64_t get_interpolation_mode() const; 1152 void set_padding_mode(const int64_t &padding_mode); 1153 int64_t get_padding_mode() const; 1154 void set_align_corners(const bool &align_corners); 1155 bool get_align_corners() const; 1156 }; 1157 1158 class MIND_API GridSampler3D : public BaseOperator { 1159 public: 1160 MIND_API_BASE_MEMBER(GridSampler3D); GridSampler3D()1161 GridSampler3D() : BaseOperator(kNameGridSampler3D) {} 1162 void set_interpolation_mode(const int64_t &interpolation_mode); 1163 int64_t get_interpolation_mode() const; 1164 void set_padding_mode(const int64_t &padding_mode); 1165 int64_t get_padding_mode() const; 1166 void set_align_corners(const bool &align_corners); 1167 bool get_align_corners() const; 1168 }; 1169 1170 class MIND_API GroupNormGrad : public BaseOperator { 1171 public: 1172 MIND_API_BASE_MEMBER(GroupNormGrad); GroupNormGrad()1173 GroupNormGrad() : BaseOperator(kNameGroupNormGrad) {} 1174 }; 1175 1176 class MIND_API GroupNorm : public BaseOperator { 1177 public: 1178 MIND_API_BASE_MEMBER(GroupNorm); GroupNorm()1179 GroupNorm() : BaseOperator(kNameGroupNorm) {} 1180 }; 1181 1182 class MIND_API HShrinkGrad : public BaseOperator { 1183 public: 1184 MIND_API_BASE_MEMBER(HShrinkGrad); HShrinkGrad()1185 HShrinkGrad() : BaseOperator(kNameHShrinkGrad) {} 1186 void set_lambd(const float &lambd); 1187 float get_lambd() const; 1188 }; 1189 1190 class MIND_API HShrink : public BaseOperator { 1191 public: 1192 MIND_API_BASE_MEMBER(HShrink); HShrink()1193 HShrink() : BaseOperator(kNameHShrink) {} 1194 void set_lambd(const float &lambd); 1195 float get_lambd() const; 1196 }; 1197 1198 class MIND_API HSigmoidGrad : public BaseOperator { 1199 public: 1200 MIND_API_BASE_MEMBER(HSigmoidGrad); HSigmoidGrad()1201 HSigmoidGrad() : BaseOperator(kNameHSigmoidGrad) {} 1202 }; 1203 1204 class MIND_API HSigmoid : public BaseOperator { 1205 public: 1206 MIND_API_BASE_MEMBER(HSigmoid); HSigmoid()1207 HSigmoid() : BaseOperator(kNameHSigmoid) {} 1208 }; 1209 1210 class MIND_API HSwishGrad : public BaseOperator { 1211 public: 1212 MIND_API_BASE_MEMBER(HSwishGrad); HSwishGrad()1213 HSwishGrad() : BaseOperator(kNameHSwishGrad) {} 1214 }; 1215 1216 class MIND_API HSwish : public BaseOperator { 1217 public: 1218 MIND_API_BASE_MEMBER(HSwish); HSwish()1219 HSwish() : BaseOperator(kNameHSwish) {} 1220 }; 1221 1222 class MIND_API Identity : public BaseOperator { 1223 public: 1224 MIND_API_BASE_MEMBER(Identity); Identity()1225 Identity() : BaseOperator(kNameIdentity) {} 1226 }; 1227 1228 class MIND_API IFFT2 : public BaseOperator { 1229 public: 1230 MIND_API_BASE_MEMBER(IFFT2); IFFT2()1231 IFFT2() : BaseOperator(kNameIFFT2) {} 1232 }; 1233 1234 class MIND_API IFFT : public BaseOperator { 1235 public: 1236 MIND_API_BASE_MEMBER(IFFT); IFFT()1237 IFFT() : BaseOperator(kNameIFFT) {} 1238 }; 1239 1240 class MIND_API IFFTN : public BaseOperator { 1241 public: 1242 MIND_API_BASE_MEMBER(IFFTN); IFFTN()1243 IFFTN() : BaseOperator(kNameIFFTN) {} 1244 }; 1245 1246 class MIND_API IFFTShift : public BaseOperator { 1247 public: 1248 MIND_API_BASE_MEMBER(IFFTShift); IFFTShift()1249 IFFTShift() : BaseOperator(kNameIFFTShift) {} 1250 }; 1251 1252 class MIND_API Im2ColExt : public BaseOperator { 1253 public: 1254 MIND_API_BASE_MEMBER(Im2ColExt); Im2ColExt()1255 Im2ColExt() : BaseOperator(kNameIm2ColExt) {} 1256 }; 1257 1258 class MIND_API IndexAddExt : public BaseOperator { 1259 public: 1260 MIND_API_BASE_MEMBER(IndexAddExt); IndexAddExt()1261 IndexAddExt() : BaseOperator(kNameIndexAddExt) {} 1262 }; 1263 1264 class MIND_API IndexSelect : public BaseOperator { 1265 public: 1266 MIND_API_BASE_MEMBER(IndexSelect); IndexSelect()1267 IndexSelect() : BaseOperator(kNameIndexSelect) {} 1268 }; 1269 1270 class MIND_API IRFFTGrad : public BaseOperator { 1271 public: 1272 MIND_API_BASE_MEMBER(IRFFTGrad); IRFFTGrad()1273 IRFFTGrad() : BaseOperator(kNameIRFFTGrad) {} 1274 }; 1275 1276 class MIND_API IRFFT : public BaseOperator { 1277 public: 1278 MIND_API_BASE_MEMBER(IRFFT); IRFFT()1279 IRFFT() : BaseOperator(kNameIRFFT) {} 1280 }; 1281 1282 class MIND_API IsClose : public BaseOperator { 1283 public: 1284 MIND_API_BASE_MEMBER(IsClose); IsClose()1285 IsClose() : BaseOperator(kNameIsClose) {} 1286 void set_rtol(const float &rtol); 1287 float get_rtol() const; 1288 void set_atol(const float &atol); 1289 float get_atol() const; 1290 void set_equal_nan(const bool &equal_nan); 1291 bool get_equal_nan() const; 1292 }; 1293 1294 class MIND_API IsFinite : public BaseOperator { 1295 public: 1296 MIND_API_BASE_MEMBER(IsFinite); IsFinite()1297 IsFinite() : BaseOperator(kNameIsFinite) {} 1298 }; 1299 1300 class MIND_API LayerNormExt : public BaseOperator { 1301 public: 1302 MIND_API_BASE_MEMBER(LayerNormExt); LayerNormExt()1303 LayerNormExt() : BaseOperator(kNameLayerNormExt) {} 1304 }; 1305 1306 class MIND_API LayerNormGradExt : public BaseOperator { 1307 public: 1308 MIND_API_BASE_MEMBER(LayerNormGradExt); LayerNormGradExt()1309 LayerNormGradExt() : BaseOperator(kNameLayerNormGradExt) {} 1310 }; 1311 1312 class MIND_API LayerNormGradGrad : public BaseOperator { 1313 public: 1314 MIND_API_BASE_MEMBER(LayerNormGradGrad); LayerNormGradGrad()1315 LayerNormGradGrad() : BaseOperator(kNameLayerNormGradGrad) {} 1316 void set_begin_norm_axis(const int64_t &begin_norm_axis); 1317 int64_t get_begin_norm_axis() const; 1318 void set_begin_params_axis(const int64_t &begin_params_axis); 1319 int64_t get_begin_params_axis() const; 1320 }; 1321 1322 class MIND_API LayerNormGrad : public BaseOperator { 1323 public: 1324 MIND_API_BASE_MEMBER(LayerNormGrad); LayerNormGrad()1325 LayerNormGrad() : BaseOperator(kNameLayerNormGrad) {} 1326 void set_begin_norm_axis(const int64_t &begin_norm_axis); 1327 int64_t get_begin_norm_axis() const; 1328 void set_begin_params_axis(const int64_t &begin_params_axis); 1329 int64_t get_begin_params_axis() const; 1330 }; 1331 1332 class MIND_API LayerNormGradV3 : public BaseOperator { 1333 public: 1334 MIND_API_BASE_MEMBER(LayerNormGradV3); LayerNormGradV3()1335 LayerNormGradV3() : BaseOperator(kNameLayerNormGradV3) {} 1336 void set_begin_norm_axis(const int64_t &begin_norm_axis); 1337 int64_t get_begin_norm_axis() const; 1338 void set_begin_params_axis(const int64_t &begin_params_axis); 1339 int64_t get_begin_params_axis() const; 1340 }; 1341 1342 class MIND_API LayerNorm : public BaseOperator { 1343 public: 1344 MIND_API_BASE_MEMBER(LayerNorm); LayerNorm()1345 LayerNorm() : BaseOperator(kNameLayerNorm) {} 1346 void set_begin_norm_axis(const int64_t &begin_norm_axis); 1347 int64_t get_begin_norm_axis() const; 1348 void set_begin_params_axis(const int64_t &begin_params_axis); 1349 int64_t get_begin_params_axis() const; 1350 void set_epsilon(const float &epsilon); 1351 float get_epsilon() const; 1352 }; 1353 1354 class MIND_API LayerNormV3 : public BaseOperator { 1355 public: 1356 MIND_API_BASE_MEMBER(LayerNormV3); LayerNormV3()1357 LayerNormV3() : BaseOperator(kNameLayerNormV3) {} 1358 void set_begin_norm_axis(const int64_t &begin_norm_axis); 1359 int64_t get_begin_norm_axis() const; 1360 void set_begin_params_axis(const int64_t &begin_params_axis); 1361 int64_t get_begin_params_axis() const; 1362 void set_epsilon(const float &epsilon); 1363 float get_epsilon() const; 1364 }; 1365 1366 class MIND_API LeakyReLUExt : public BaseOperator { 1367 public: 1368 MIND_API_BASE_MEMBER(LeakyReLUExt); LeakyReLUExt()1369 LeakyReLUExt() : BaseOperator(kNameLeakyReLUExt) {} 1370 }; 1371 1372 class MIND_API LeakyReLUGradExt : public BaseOperator { 1373 public: 1374 MIND_API_BASE_MEMBER(LeakyReLUGradExt); LeakyReLUGradExt()1375 LeakyReLUGradExt() : BaseOperator(kNameLeakyReLUGradExt) {} 1376 }; 1377 1378 class MIND_API LessEqual : public BaseOperator { 1379 public: 1380 MIND_API_BASE_MEMBER(LessEqual); LessEqual()1381 LessEqual() : BaseOperator(kNameLessEqual) {} 1382 }; 1383 1384 class MIND_API Less : public BaseOperator { 1385 public: 1386 MIND_API_BASE_MEMBER(Less); Less()1387 Less() : BaseOperator(kNameLess) {} 1388 }; 1389 1390 class MIND_API LinSpaceExt : public BaseOperator { 1391 public: 1392 MIND_API_BASE_MEMBER(LinSpaceExt); LinSpaceExt()1393 LinSpaceExt() : BaseOperator(kNameLinSpaceExt) {} 1394 }; 1395 1396 class MIND_API LinSpace : public BaseOperator { 1397 public: 1398 MIND_API_BASE_MEMBER(LinSpace); LinSpace()1399 LinSpace() : BaseOperator(kNameLinSpace) {} 1400 }; 1401 1402 class MIND_API ListToTuple : public BaseOperator { 1403 public: 1404 MIND_API_BASE_MEMBER(ListToTuple); ListToTuple()1405 ListToTuple() : BaseOperator(kNameListToTuple) {} 1406 }; 1407 1408 class MIND_API Log1p : public BaseOperator { 1409 public: 1410 MIND_API_BASE_MEMBER(Log1p); Log1p()1411 Log1p() : BaseOperator(kNameLog1p) {} 1412 }; 1413 1414 class MIND_API LogMatrixDeterminant : public BaseOperator { 1415 public: 1416 MIND_API_BASE_MEMBER(LogMatrixDeterminant); LogMatrixDeterminant()1417 LogMatrixDeterminant() : BaseOperator(kNameLogMatrixDeterminant) {} 1418 }; 1419 1420 class MIND_API Log : public BaseOperator { 1421 public: 1422 MIND_API_BASE_MEMBER(Log); Log()1423 Log() : BaseOperator(kNameLog) {} 1424 }; 1425 1426 class MIND_API LogSoftmaxGrad : public BaseOperator { 1427 public: 1428 MIND_API_BASE_MEMBER(LogSoftmaxGrad); LogSoftmaxGrad()1429 LogSoftmaxGrad() : BaseOperator(kNameLogSoftmaxGrad) {} 1430 void set_axis(const int64_t &axis); 1431 int64_t get_axis() const; 1432 }; 1433 1434 class MIND_API LogSoftmax : public BaseOperator { 1435 public: 1436 MIND_API_BASE_MEMBER(LogSoftmax); LogSoftmax()1437 LogSoftmax() : BaseOperator(kNameLogSoftmax) {} 1438 void set_axis(const int64_t &axis); 1439 int64_t get_axis() const; 1440 }; 1441 1442 class MIND_API LogicalAnd : public BaseOperator { 1443 public: 1444 MIND_API_BASE_MEMBER(LogicalAnd); LogicalAnd()1445 LogicalAnd() : BaseOperator(kNameLogicalAnd) {} 1446 }; 1447 1448 class MIND_API LogicalNot : public BaseOperator { 1449 public: 1450 MIND_API_BASE_MEMBER(LogicalNot); LogicalNot()1451 LogicalNot() : BaseOperator(kNameLogicalNot) {} 1452 }; 1453 1454 class MIND_API LogicalOr : public BaseOperator { 1455 public: 1456 MIND_API_BASE_MEMBER(LogicalOr); LogicalOr()1457 LogicalOr() : BaseOperator(kNameLogicalOr) {} 1458 }; 1459 1460 class MIND_API LogicalXor : public BaseOperator { 1461 public: 1462 MIND_API_BASE_MEMBER(LogicalXor); LogicalXor()1463 LogicalXor() : BaseOperator(kNameLogicalXor) {} 1464 }; 1465 1466 class MIND_API LogitGrad : public BaseOperator { 1467 public: 1468 MIND_API_BASE_MEMBER(LogitGrad); LogitGrad()1469 LogitGrad() : BaseOperator(kNameLogitGrad) {} 1470 void set_eps(const float &eps); 1471 float get_eps() const; 1472 }; 1473 1474 class MIND_API Logit : public BaseOperator { 1475 public: 1476 MIND_API_BASE_MEMBER(Logit); Logit()1477 Logit() : BaseOperator(kNameLogit) {} 1478 void set_eps(const float &eps); 1479 float get_eps() const; 1480 }; 1481 1482 class MIND_API MaskedFill : public BaseOperator { 1483 public: 1484 MIND_API_BASE_MEMBER(MaskedFill); MaskedFill()1485 MaskedFill() : BaseOperator(kNameMaskedFill) {} 1486 }; 1487 1488 class MIND_API MatMulExt : public BaseOperator { 1489 public: 1490 MIND_API_BASE_MEMBER(MatMulExt); MatMulExt()1491 MatMulExt() : BaseOperator(kNameMatMulExt) {} 1492 }; 1493 1494 class MIND_API MatMul : public BaseOperator { 1495 public: 1496 MIND_API_BASE_MEMBER(MatMul); MatMul()1497 MatMul() : BaseOperator(kNameMatMul) {} 1498 void set_transpose_a(const bool &transpose_a); 1499 bool get_transpose_a() const; 1500 void set_transpose_b(const bool &transpose_b); 1501 bool get_transpose_b() const; 1502 }; 1503 1504 class MIND_API MatrixDeterminant : public BaseOperator { 1505 public: 1506 MIND_API_BASE_MEMBER(MatrixDeterminant); MatrixDeterminant()1507 MatrixDeterminant() : BaseOperator(kNameMatrixDeterminant) {} 1508 }; 1509 1510 class MIND_API MatrixExp : public BaseOperator { 1511 public: 1512 MIND_API_BASE_MEMBER(MatrixExp); MatrixExp()1513 MatrixExp() : BaseOperator(kNameMatrixExp) {} 1514 }; 1515 1516 class MIND_API MatrixInverseExt : public BaseOperator { 1517 public: 1518 MIND_API_BASE_MEMBER(MatrixInverseExt); MatrixInverseExt()1519 MatrixInverseExt() : BaseOperator(kNameMatrixInverseExt) {} 1520 }; 1521 1522 class MIND_API Max : public BaseOperator { 1523 public: 1524 MIND_API_BASE_MEMBER(Max); Max()1525 Max() : BaseOperator(kNameMax) {} 1526 }; 1527 1528 class MIND_API MaxPoolGradWithIndices : public BaseOperator { 1529 public: 1530 MIND_API_BASE_MEMBER(MaxPoolGradWithIndices); MaxPoolGradWithIndices()1531 MaxPoolGradWithIndices() : BaseOperator(kNameMaxPoolGradWithIndices) {} 1532 void set_kernel_size(const std::vector<int64_t> &kernel_size); 1533 std::vector<int64_t> get_kernel_size() const; 1534 void set_strides(const std::vector<int64_t> &strides); 1535 std::vector<int64_t> get_strides() const; 1536 void set_pads(const std::vector<int64_t> &pads); 1537 std::vector<int64_t> get_pads() const; 1538 void set_dilation(const std::vector<int64_t> &dilation); 1539 std::vector<int64_t> get_dilation() const; 1540 void set_ceil_mode(const bool &ceil_mode); 1541 bool get_ceil_mode() const; 1542 void set_argmax_type(const int64_t &argmax_type); 1543 int64_t get_argmax_type() const; 1544 }; 1545 1546 class MIND_API MaxPoolGradWithMask : public BaseOperator { 1547 public: 1548 MIND_API_BASE_MEMBER(MaxPoolGradWithMask); MaxPoolGradWithMask()1549 MaxPoolGradWithMask() : BaseOperator(kNameMaxPoolGradWithMask) {} 1550 void set_kernel_size(const std::vector<int64_t> &kernel_size); 1551 std::vector<int64_t> get_kernel_size() const; 1552 void set_strides(const std::vector<int64_t> &strides); 1553 std::vector<int64_t> get_strides() const; 1554 void set_pads(const std::vector<int64_t> &pads); 1555 std::vector<int64_t> get_pads() const; 1556 void set_dilation(const std::vector<int64_t> &dilation); 1557 std::vector<int64_t> get_dilation() const; 1558 void set_ceil_mode(const bool &ceil_mode); 1559 bool get_ceil_mode() const; 1560 void set_argmax_type(const int64_t &argmax_type); 1561 int64_t get_argmax_type() const; 1562 }; 1563 1564 class MIND_API MaxPoolWithIndices : public BaseOperator { 1565 public: 1566 MIND_API_BASE_MEMBER(MaxPoolWithIndices); MaxPoolWithIndices()1567 MaxPoolWithIndices() : BaseOperator(kNameMaxPoolWithIndices) {} 1568 void set_kernel_size(const std::vector<int64_t> &kernel_size); 1569 std::vector<int64_t> get_kernel_size() const; 1570 void set_strides(const std::vector<int64_t> &strides); 1571 std::vector<int64_t> get_strides() const; 1572 void set_pads(const std::vector<int64_t> &pads); 1573 std::vector<int64_t> get_pads() const; 1574 void set_dilation(const std::vector<int64_t> &dilation); 1575 std::vector<int64_t> get_dilation() const; 1576 void set_ceil_mode(const bool &ceil_mode); 1577 bool get_ceil_mode() const; 1578 void set_argmax_type(const int64_t &argmax_type); 1579 int64_t get_argmax_type() const; 1580 }; 1581 1582 class MIND_API MaxPoolWithMask : public BaseOperator { 1583 public: 1584 MIND_API_BASE_MEMBER(MaxPoolWithMask); MaxPoolWithMask()1585 MaxPoolWithMask() : BaseOperator(kNameMaxPoolWithMask) {} 1586 void set_kernel_size(const std::vector<int64_t> &kernel_size); 1587 std::vector<int64_t> get_kernel_size() const; 1588 void set_strides(const std::vector<int64_t> &strides); 1589 std::vector<int64_t> get_strides() const; 1590 void set_pads(const std::vector<int64_t> &pads); 1591 std::vector<int64_t> get_pads() const; 1592 void set_dilation(const std::vector<int64_t> &dilation); 1593 std::vector<int64_t> get_dilation() const; 1594 void set_ceil_mode(const bool &ceil_mode); 1595 bool get_ceil_mode() const; 1596 void set_argmax_type(const int64_t &argmax_type); 1597 int64_t get_argmax_type() const; 1598 }; 1599 1600 class MIND_API MaximumGradGrad : public BaseOperator { 1601 public: 1602 MIND_API_BASE_MEMBER(MaximumGradGrad); MaximumGradGrad()1603 MaximumGradGrad() : BaseOperator(kNameMaximumGradGrad) {} 1604 void set_grad_x(const bool &grad_x); 1605 bool get_grad_x() const; 1606 void set_grad_y(const bool &grad_y); 1607 bool get_grad_y() const; 1608 }; 1609 1610 class MIND_API MaximumGrad : public BaseOperator { 1611 public: 1612 MIND_API_BASE_MEMBER(MaximumGrad); MaximumGrad()1613 MaximumGrad() : BaseOperator(kNameMaximumGrad) {} 1614 void set_grad_x(const bool &grad_x); 1615 bool get_grad_x() const; 1616 void set_grad_y(const bool &grad_y); 1617 bool get_grad_y() const; 1618 }; 1619 1620 class MIND_API Maximum : public BaseOperator { 1621 public: 1622 MIND_API_BASE_MEMBER(Maximum); Maximum()1623 Maximum() : BaseOperator(kNameMaximum) {} 1624 }; 1625 1626 class MIND_API MeanExt : public BaseOperator { 1627 public: 1628 MIND_API_BASE_MEMBER(MeanExt); MeanExt()1629 MeanExt() : BaseOperator(kNameMeanExt) {} 1630 }; 1631 1632 class MIND_API Min : public BaseOperator { 1633 public: 1634 MIND_API_BASE_MEMBER(Min); Min()1635 Min() : BaseOperator(kNameMin) {} 1636 }; 1637 1638 class MIND_API MinimumGrad : public BaseOperator { 1639 public: 1640 MIND_API_BASE_MEMBER(MinimumGrad); MinimumGrad()1641 MinimumGrad() : BaseOperator(kNameMinimumGrad) {} 1642 void set_grad_x(const bool &grad_x); 1643 bool get_grad_x() const; 1644 void set_grad_y(const bool &grad_y); 1645 bool get_grad_y() const; 1646 }; 1647 1648 class MIND_API Minimum : public BaseOperator { 1649 public: 1650 MIND_API_BASE_MEMBER(Minimum); Minimum()1651 Minimum() : BaseOperator(kNameMinimum) {} 1652 }; 1653 1654 class MIND_API Mul : public BaseOperator { 1655 public: 1656 MIND_API_BASE_MEMBER(Mul); Mul()1657 Mul() : BaseOperator(kNameMul) {} 1658 }; 1659 1660 class MIND_API Mv : public BaseOperator { 1661 public: 1662 MIND_API_BASE_MEMBER(Mv); Mv()1663 Mv() : BaseOperator(kNameMv) {} 1664 }; 1665 1666 class MIND_API NanToNum : public BaseOperator { 1667 public: 1668 MIND_API_BASE_MEMBER(NanToNum); NanToNum()1669 NanToNum() : BaseOperator(kNameNanToNum) {} 1670 void set_nan(const float &nan); 1671 float get_nan() const; 1672 void set_posinf(const float &posinf); 1673 float get_posinf() const; 1674 void set_neginf(const float &neginf); 1675 float get_neginf() const; 1676 }; 1677 1678 class MIND_API Neg : public BaseOperator { 1679 public: 1680 MIND_API_BASE_MEMBER(Neg); Neg()1681 Neg() : BaseOperator(kNameNeg) {} 1682 }; 1683 1684 class MIND_API NextAfter : public BaseOperator { 1685 public: 1686 MIND_API_BASE_MEMBER(NextAfter); NextAfter()1687 NextAfter() : BaseOperator(kNameNextAfter) {} 1688 }; 1689 1690 class MIND_API NLLLossGrad : public BaseOperator { 1691 public: 1692 MIND_API_BASE_MEMBER(NLLLossGrad); NLLLossGrad()1693 NLLLossGrad() : BaseOperator(kNameNLLLossGrad) {} 1694 void set_reduction(const int64_t &reduction); 1695 int64_t get_reduction() const; 1696 void set_ignore_index(const int64_t &ignore_index); 1697 int64_t get_ignore_index() const; 1698 }; 1699 1700 class MIND_API NLLLoss : public BaseOperator { 1701 public: 1702 MIND_API_BASE_MEMBER(NLLLoss); NLLLoss()1703 NLLLoss() : BaseOperator(kNameNLLLoss) {} 1704 void set_reduction(const int64_t &reduction); 1705 int64_t get_reduction() const; 1706 void set_ignore_index(const int64_t &ignore_index); 1707 int64_t get_ignore_index() const; 1708 }; 1709 1710 class MIND_API NonZeroExt : public BaseOperator { 1711 public: 1712 MIND_API_BASE_MEMBER(NonZeroExt); NonZeroExt()1713 NonZeroExt() : BaseOperator(kNameNonZeroExt) {} 1714 }; 1715 1716 class MIND_API NonZero : public BaseOperator { 1717 public: 1718 MIND_API_BASE_MEMBER(NonZero); NonZero()1719 NonZero() : BaseOperator(kNameNonZero) {} 1720 }; 1721 1722 class MIND_API Norm : public BaseOperator { 1723 public: 1724 MIND_API_BASE_MEMBER(Norm); Norm()1725 Norm() : BaseOperator(kNameNorm) {} 1726 }; 1727 1728 class MIND_API NormalFloatFloat : public BaseOperator { 1729 public: 1730 MIND_API_BASE_MEMBER(NormalFloatFloat); NormalFloatFloat()1731 NormalFloatFloat() : BaseOperator(kNameNormalFloatFloat) {} 1732 }; 1733 1734 class MIND_API NormalFloatTensor : public BaseOperator { 1735 public: 1736 MIND_API_BASE_MEMBER(NormalFloatTensor); NormalFloatTensor()1737 NormalFloatTensor() : BaseOperator(kNameNormalFloatTensor) {} 1738 }; 1739 1740 class MIND_API NormalTensorFloat : public BaseOperator { 1741 public: 1742 MIND_API_BASE_MEMBER(NormalTensorFloat); NormalTensorFloat()1743 NormalTensorFloat() : BaseOperator(kNameNormalTensorFloat) {} 1744 }; 1745 1746 class MIND_API NormalTensorTensor : public BaseOperator { 1747 public: 1748 MIND_API_BASE_MEMBER(NormalTensorTensor); NormalTensorTensor()1749 NormalTensorTensor() : BaseOperator(kNameNormalTensorTensor) {} 1750 }; 1751 1752 class MIND_API NotEqual : public BaseOperator { 1753 public: 1754 MIND_API_BASE_MEMBER(NotEqual); NotEqual()1755 NotEqual() : BaseOperator(kNameNotEqual) {} 1756 }; 1757 1758 class MIND_API NPUClearFloatStatusV2 : public BaseOperator { 1759 public: 1760 MIND_API_BASE_MEMBER(NPUClearFloatStatusV2); NPUClearFloatStatusV2()1761 NPUClearFloatStatusV2() : BaseOperator(kNameNPUClearFloatStatusV2) {} 1762 }; 1763 1764 class MIND_API NPUGetFloatStatusV2 : public BaseOperator { 1765 public: 1766 MIND_API_BASE_MEMBER(NPUGetFloatStatusV2); NPUGetFloatStatusV2()1767 NPUGetFloatStatusV2() : BaseOperator(kNameNPUGetFloatStatusV2) {} 1768 }; 1769 1770 class MIND_API OneHotExt : public BaseOperator { 1771 public: 1772 MIND_API_BASE_MEMBER(OneHotExt); OneHotExt()1773 OneHotExt() : BaseOperator(kNameOneHotExt) {} 1774 void set_axis(const int64_t &axis); 1775 int64_t get_axis() const; 1776 }; 1777 1778 class MIND_API OneHot : public BaseOperator { 1779 public: 1780 MIND_API_BASE_MEMBER(OneHot); OneHot()1781 OneHot() : BaseOperator(kNameOneHot) {} 1782 void set_axis(const int64_t &axis); 1783 int64_t get_axis() const; 1784 }; 1785 1786 class MIND_API OnesLikeExt : public BaseOperator { 1787 public: 1788 MIND_API_BASE_MEMBER(OnesLikeExt); OnesLikeExt()1789 OnesLikeExt() : BaseOperator(kNameOnesLikeExt) {} 1790 }; 1791 1792 class MIND_API OnesLike : public BaseOperator { 1793 public: 1794 MIND_API_BASE_MEMBER(OnesLike); OnesLike()1795 OnesLike() : BaseOperator(kNameOnesLike) {} 1796 }; 1797 1798 class MIND_API Ones : public BaseOperator { 1799 public: 1800 MIND_API_BASE_MEMBER(Ones); Ones()1801 Ones() : BaseOperator(kNameOnes) {} 1802 }; 1803 1804 class MIND_API PagedAttentionMask : public BaseOperator { 1805 public: 1806 MIND_API_BASE_MEMBER(PagedAttentionMask); PagedAttentionMask()1807 PagedAttentionMask() : BaseOperator(kNamePagedAttentionMask) {} 1808 void set_head_num(const int64_t &head_num); 1809 int64_t get_head_num() const; 1810 void set_scale_value(const float &scale_value); 1811 float get_scale_value() const; 1812 void set_kv_head_num(const int64_t &kv_head_num); 1813 int64_t get_kv_head_num() const; 1814 }; 1815 1816 class MIND_API PagedAttention : public BaseOperator { 1817 public: 1818 MIND_API_BASE_MEMBER(PagedAttention); PagedAttention()1819 PagedAttention() : BaseOperator(kNamePagedAttention) {} 1820 void set_head_num(const int64_t &head_num); 1821 int64_t get_head_num() const; 1822 void set_scale_value(const float &scale_value); 1823 float get_scale_value() const; 1824 void set_kv_head_num(const int64_t &kv_head_num); 1825 int64_t get_kv_head_num() const; 1826 }; 1827 1828 class MIND_API Pow : public BaseOperator { 1829 public: 1830 MIND_API_BASE_MEMBER(Pow); Pow()1831 Pow() : BaseOperator(kNamePow) {} 1832 }; 1833 1834 class MIND_API PReLUGrad : public BaseOperator { 1835 public: 1836 MIND_API_BASE_MEMBER(PReLUGrad); PReLUGrad()1837 PReLUGrad() : BaseOperator(kNamePReLUGrad) {} 1838 }; 1839 1840 class MIND_API PReLU : public BaseOperator { 1841 public: 1842 MIND_API_BASE_MEMBER(PReLU); PReLU()1843 PReLU() : BaseOperator(kNamePReLU) {} 1844 }; 1845 1846 class MIND_API ProdExt : public BaseOperator { 1847 public: 1848 MIND_API_BASE_MEMBER(ProdExt); ProdExt()1849 ProdExt() : BaseOperator(kNameProdExt) {} 1850 }; 1851 1852 class MIND_API PromptKVCache : public BaseOperator { 1853 public: 1854 MIND_API_BASE_MEMBER(PromptKVCache); PromptKVCache()1855 PromptKVCache() : BaseOperator(kNamePromptKVCache) {} 1856 void set_align_mode(const int64_t &align_mode); 1857 int64_t get_align_mode() const; 1858 }; 1859 1860 class MIND_API Qr : public BaseOperator { 1861 public: 1862 MIND_API_BASE_MEMBER(Qr); Qr()1863 Qr() : BaseOperator(kNameQr) {} 1864 void set_full_matrices(const bool &full_matrices); 1865 bool get_full_matrices() const; 1866 }; 1867 1868 class MIND_API RandExt : public BaseOperator { 1869 public: 1870 MIND_API_BASE_MEMBER(RandExt); RandExt()1871 RandExt() : BaseOperator(kNameRandExt) {} 1872 }; 1873 1874 class MIND_API RandLikeExt : public BaseOperator { 1875 public: 1876 MIND_API_BASE_MEMBER(RandLikeExt); RandLikeExt()1877 RandLikeExt() : BaseOperator(kNameRandLikeExt) {} 1878 }; 1879 1880 class MIND_API RandpermV2 : public BaseOperator { 1881 public: 1882 MIND_API_BASE_MEMBER(RandpermV2); RandpermV2()1883 RandpermV2() : BaseOperator(kNameRandpermV2) {} 1884 void set_seed(const int64_t &seed); 1885 int64_t get_seed() const; 1886 void set_offset(const int64_t &offset); 1887 int64_t get_offset() const; 1888 void set_dtype(const int64_t &dtype); 1889 int64_t get_dtype() const; 1890 }; 1891 1892 class MIND_API Range : public BaseOperator { 1893 public: 1894 MIND_API_BASE_MEMBER(Range); Range()1895 Range() : BaseOperator(kNameRange) {} 1896 void set_maxlen(const int64_t &maxlen); 1897 int64_t get_maxlen() const; 1898 }; 1899 1900 class MIND_API Rank : public BaseOperator { 1901 public: 1902 MIND_API_BASE_MEMBER(Rank); Rank()1903 Rank() : BaseOperator(kNameRank) {} 1904 }; 1905 1906 class MIND_API RealDiv : public BaseOperator { 1907 public: 1908 MIND_API_BASE_MEMBER(RealDiv); RealDiv()1909 RealDiv() : BaseOperator(kNameRealDiv) {} 1910 }; 1911 1912 class MIND_API Real : public BaseOperator { 1913 public: 1914 MIND_API_BASE_MEMBER(Real); Real()1915 Real() : BaseOperator(kNameReal) {} 1916 }; 1917 1918 class MIND_API ReciprocalGrad : public BaseOperator { 1919 public: 1920 MIND_API_BASE_MEMBER(ReciprocalGrad); ReciprocalGrad()1921 ReciprocalGrad() : BaseOperator(kNameReciprocalGrad) {} 1922 }; 1923 1924 class MIND_API Reciprocal : public BaseOperator { 1925 public: 1926 MIND_API_BASE_MEMBER(Reciprocal); Reciprocal()1927 Reciprocal() : BaseOperator(kNameReciprocal) {} 1928 }; 1929 1930 class MIND_API ReduceAll : public BaseOperator { 1931 public: 1932 MIND_API_BASE_MEMBER(ReduceAll); ReduceAll()1933 ReduceAll() : BaseOperator(kNameReduceAll) {} 1934 void set_keep_dims(const bool &keep_dims); 1935 bool get_keep_dims() const; 1936 }; 1937 1938 class MIND_API ReduceAny : public BaseOperator { 1939 public: 1940 MIND_API_BASE_MEMBER(ReduceAny); ReduceAny()1941 ReduceAny() : BaseOperator(kNameReduceAny) {} 1942 void set_keep_dims(const bool &keep_dims); 1943 bool get_keep_dims() const; 1944 }; 1945 1946 class MIND_API ReduceMax : public BaseOperator { 1947 public: 1948 MIND_API_BASE_MEMBER(ReduceMax); ReduceMax()1949 ReduceMax() : BaseOperator(kNameReduceMax) {} 1950 void set_keep_dims(const bool &keep_dims); 1951 bool get_keep_dims() const; 1952 }; 1953 1954 class MIND_API ReduceMean : public BaseOperator { 1955 public: 1956 MIND_API_BASE_MEMBER(ReduceMean); ReduceMean()1957 ReduceMean() : BaseOperator(kNameReduceMean) {} 1958 void set_keep_dims(const bool &keep_dims); 1959 bool get_keep_dims() const; 1960 }; 1961 1962 class MIND_API ReduceMin : public BaseOperator { 1963 public: 1964 MIND_API_BASE_MEMBER(ReduceMin); ReduceMin()1965 ReduceMin() : BaseOperator(kNameReduceMin) {} 1966 void set_keep_dims(const bool &keep_dims); 1967 bool get_keep_dims() const; 1968 }; 1969 1970 class MIND_API ReduceProd : public BaseOperator { 1971 public: 1972 MIND_API_BASE_MEMBER(ReduceProd); ReduceProd()1973 ReduceProd() : BaseOperator(kNameReduceProd) {} 1974 void set_keep_dims(const bool &keep_dims); 1975 bool get_keep_dims() const; 1976 }; 1977 1978 class MIND_API ReduceStd : public BaseOperator { 1979 public: 1980 MIND_API_BASE_MEMBER(ReduceStd); ReduceStd()1981 ReduceStd() : BaseOperator(kNameReduceStd) {} 1982 void set_axis(const std::vector<int64_t> &axis); 1983 std::vector<int64_t> get_axis() const; 1984 void set_unbiased(const bool &unbiased); 1985 bool get_unbiased() const; 1986 void set_keep_dims(const bool &keep_dims); 1987 bool get_keep_dims() const; 1988 }; 1989 1990 class MIND_API ReduceSum : public BaseOperator { 1991 public: 1992 MIND_API_BASE_MEMBER(ReduceSum); ReduceSum()1993 ReduceSum() : BaseOperator(kNameReduceSum) {} 1994 void set_keep_dims(const bool &keep_dims); 1995 bool get_keep_dims() const; 1996 void set_skip_mode(const bool &skip_mode); 1997 bool get_skip_mode() const; 1998 }; 1999 2000 class MIND_API ReflectionPad1DGrad : public BaseOperator { 2001 public: 2002 MIND_API_BASE_MEMBER(ReflectionPad1DGrad); ReflectionPad1DGrad()2003 ReflectionPad1DGrad() : BaseOperator(kNameReflectionPad1DGrad) {} 2004 }; 2005 2006 class MIND_API ReflectionPad1D : public BaseOperator { 2007 public: 2008 MIND_API_BASE_MEMBER(ReflectionPad1D); ReflectionPad1D()2009 ReflectionPad1D() : BaseOperator(kNameReflectionPad1D) {} 2010 }; 2011 2012 class MIND_API ReflectionPad2DGrad : public BaseOperator { 2013 public: 2014 MIND_API_BASE_MEMBER(ReflectionPad2DGrad); ReflectionPad2DGrad()2015 ReflectionPad2DGrad() : BaseOperator(kNameReflectionPad2DGrad) {} 2016 }; 2017 2018 class MIND_API ReflectionPad2D : public BaseOperator { 2019 public: 2020 MIND_API_BASE_MEMBER(ReflectionPad2D); ReflectionPad2D()2021 ReflectionPad2D() : BaseOperator(kNameReflectionPad2D) {} 2022 }; 2023 2024 class MIND_API ReflectionPad3DGrad : public BaseOperator { 2025 public: 2026 MIND_API_BASE_MEMBER(ReflectionPad3DGrad); ReflectionPad3DGrad()2027 ReflectionPad3DGrad() : BaseOperator(kNameReflectionPad3DGrad) {} 2028 }; 2029 2030 class MIND_API ReflectionPad3D : public BaseOperator { 2031 public: 2032 MIND_API_BASE_MEMBER(ReflectionPad3D); ReflectionPad3D()2033 ReflectionPad3D() : BaseOperator(kNameReflectionPad3D) {} 2034 }; 2035 2036 class MIND_API ReLU6Grad : public BaseOperator { 2037 public: 2038 MIND_API_BASE_MEMBER(ReLU6Grad); ReLU6Grad()2039 ReLU6Grad() : BaseOperator(kNameReLU6Grad) {} 2040 }; 2041 2042 class MIND_API ReLU6 : public BaseOperator { 2043 public: 2044 MIND_API_BASE_MEMBER(ReLU6); ReLU6()2045 ReLU6() : BaseOperator(kNameReLU6) {} 2046 }; 2047 2048 class MIND_API ReluGrad : public BaseOperator { 2049 public: 2050 MIND_API_BASE_MEMBER(ReluGrad); ReluGrad()2051 ReluGrad() : BaseOperator(kNameReluGrad) {} 2052 }; 2053 2054 class MIND_API ReLU : public BaseOperator { 2055 public: 2056 MIND_API_BASE_MEMBER(ReLU); ReLU()2057 ReLU() : BaseOperator(kNameReLU) {} 2058 }; 2059 2060 class MIND_API RepeatInterleaveGrad : public BaseOperator { 2061 public: 2062 MIND_API_BASE_MEMBER(RepeatInterleaveGrad); RepeatInterleaveGrad()2063 RepeatInterleaveGrad() : BaseOperator(kNameRepeatInterleaveGrad) {} 2064 }; 2065 2066 class MIND_API RepeatInterleaveInt : public BaseOperator { 2067 public: 2068 MIND_API_BASE_MEMBER(RepeatInterleaveInt); RepeatInterleaveInt()2069 RepeatInterleaveInt() : BaseOperator(kNameRepeatInterleaveInt) {} 2070 }; 2071 2072 class MIND_API RepeatInterleaveTensor : public BaseOperator { 2073 public: 2074 MIND_API_BASE_MEMBER(RepeatInterleaveTensor); RepeatInterleaveTensor()2075 RepeatInterleaveTensor() : BaseOperator(kNameRepeatInterleaveTensor) {} 2076 }; 2077 2078 class MIND_API ReplicationPad1DGrad : public BaseOperator { 2079 public: 2080 MIND_API_BASE_MEMBER(ReplicationPad1DGrad); ReplicationPad1DGrad()2081 ReplicationPad1DGrad() : BaseOperator(kNameReplicationPad1DGrad) {} 2082 }; 2083 2084 class MIND_API ReplicationPad1D : public BaseOperator { 2085 public: 2086 MIND_API_BASE_MEMBER(ReplicationPad1D); ReplicationPad1D()2087 ReplicationPad1D() : BaseOperator(kNameReplicationPad1D) {} 2088 }; 2089 2090 class MIND_API ReplicationPad2DGrad : public BaseOperator { 2091 public: 2092 MIND_API_BASE_MEMBER(ReplicationPad2DGrad); ReplicationPad2DGrad()2093 ReplicationPad2DGrad() : BaseOperator(kNameReplicationPad2DGrad) {} 2094 }; 2095 2096 class MIND_API ReplicationPad2D : public BaseOperator { 2097 public: 2098 MIND_API_BASE_MEMBER(ReplicationPad2D); ReplicationPad2D()2099 ReplicationPad2D() : BaseOperator(kNameReplicationPad2D) {} 2100 }; 2101 2102 class MIND_API ReplicationPad3DGrad : public BaseOperator { 2103 public: 2104 MIND_API_BASE_MEMBER(ReplicationPad3DGrad); ReplicationPad3DGrad()2105 ReplicationPad3DGrad() : BaseOperator(kNameReplicationPad3DGrad) {} 2106 }; 2107 2108 class MIND_API ReplicationPad3D : public BaseOperator { 2109 public: 2110 MIND_API_BASE_MEMBER(ReplicationPad3D); ReplicationPad3D()2111 ReplicationPad3D() : BaseOperator(kNameReplicationPad3D) {} 2112 }; 2113 2114 class MIND_API ReshapeAndCache : public BaseOperator { 2115 public: 2116 MIND_API_BASE_MEMBER(ReshapeAndCache); ReshapeAndCache()2117 ReshapeAndCache() : BaseOperator(kNameReshapeAndCache) {} 2118 }; 2119 2120 class MIND_API Reshape : public BaseOperator { 2121 public: 2122 MIND_API_BASE_MEMBER(Reshape); Reshape()2123 Reshape() : BaseOperator(kNameReshape) {} 2124 }; 2125 2126 class MIND_API ResizeBicubicGrad : public BaseOperator { 2127 public: 2128 MIND_API_BASE_MEMBER(ResizeBicubicGrad); ResizeBicubicGrad()2129 ResizeBicubicGrad() : BaseOperator(kNameResizeBicubicGrad) {} 2130 void set_align_corners(const bool &align_corners); 2131 bool get_align_corners() const; 2132 void set_half_pixel_centers(const bool &half_pixel_centers); 2133 bool get_half_pixel_centers() const; 2134 }; 2135 2136 class MIND_API ResizeBicubic : public BaseOperator { 2137 public: 2138 MIND_API_BASE_MEMBER(ResizeBicubic); ResizeBicubic()2139 ResizeBicubic() : BaseOperator(kNameResizeBicubic) {} 2140 void set_align_corners(const bool &align_corners); 2141 bool get_align_corners() const; 2142 void set_half_pixel_centers(const bool &half_pixel_centers); 2143 bool get_half_pixel_centers() const; 2144 }; 2145 2146 class MIND_API ResizeBilinearGrad : public BaseOperator { 2147 public: 2148 MIND_API_BASE_MEMBER(ResizeBilinearGrad); ResizeBilinearGrad()2149 ResizeBilinearGrad() : BaseOperator(kNameResizeBilinearGrad) {} 2150 void set_align_corners(const bool &align_corners); 2151 bool get_align_corners() const; 2152 void set_half_pixel_centers(const bool &half_pixel_centers); 2153 bool get_half_pixel_centers() const; 2154 }; 2155 2156 class MIND_API ResizeBilinearV2 : public BaseOperator { 2157 public: 2158 MIND_API_BASE_MEMBER(ResizeBilinearV2); ResizeBilinearV2()2159 ResizeBilinearV2() : BaseOperator(kNameResizeBilinearV2) {} 2160 void set_align_corners(const bool &align_corners); 2161 bool get_align_corners() const; 2162 void set_half_pixel_centers(const bool &half_pixel_centers); 2163 bool get_half_pixel_centers() const; 2164 }; 2165 2166 class MIND_API ResizeD : public BaseOperator { 2167 public: 2168 MIND_API_BASE_MEMBER(ResizeD); ResizeD()2169 ResizeD() : BaseOperator(kNameResizeD) {} 2170 }; 2171 2172 class MIND_API ResizeLinear1DGrad : public BaseOperator { 2173 public: 2174 MIND_API_BASE_MEMBER(ResizeLinear1DGrad); ResizeLinear1DGrad()2175 ResizeLinear1DGrad() : BaseOperator(kNameResizeLinear1DGrad) {} 2176 void set_coordinate_transformation_mode(const int64_t &coordinate_transformation_mode); 2177 int64_t get_coordinate_transformation_mode() const; 2178 }; 2179 2180 class MIND_API ResizeLinear1D : public BaseOperator { 2181 public: 2182 MIND_API_BASE_MEMBER(ResizeLinear1D); ResizeLinear1D()2183 ResizeLinear1D() : BaseOperator(kNameResizeLinear1D) {} 2184 void set_coordinate_transformation_mode(const int64_t &coordinate_transformation_mode); 2185 int64_t get_coordinate_transformation_mode() const; 2186 }; 2187 2188 class MIND_API ResizeNearestNeighborGrad : public BaseOperator { 2189 public: 2190 MIND_API_BASE_MEMBER(ResizeNearestNeighborGrad); ResizeNearestNeighborGrad()2191 ResizeNearestNeighborGrad() : BaseOperator(kNameResizeNearestNeighborGrad) {} 2192 void set_align_corners(const bool &align_corners); 2193 bool get_align_corners() const; 2194 void set_half_pixel_centers(const bool &half_pixel_centers); 2195 bool get_half_pixel_centers() const; 2196 }; 2197 2198 class MIND_API ResizeNearestNeighbor : public BaseOperator { 2199 public: 2200 MIND_API_BASE_MEMBER(ResizeNearestNeighbor); ResizeNearestNeighbor()2201 ResizeNearestNeighbor() : BaseOperator(kNameResizeNearestNeighbor) {} 2202 void set_size(const std::vector<int64_t> &size); 2203 std::vector<int64_t> get_size() const; 2204 void set_align_corners(const bool &align_corners); 2205 bool get_align_corners() const; 2206 void set_half_pixel_centers(const bool &half_pixel_centers); 2207 bool get_half_pixel_centers() const; 2208 }; 2209 2210 class MIND_API ResizeNearestNeighborV2Grad : public BaseOperator { 2211 public: 2212 MIND_API_BASE_MEMBER(ResizeNearestNeighborV2Grad); ResizeNearestNeighborV2Grad()2213 ResizeNearestNeighborV2Grad() : BaseOperator(kNameResizeNearestNeighborV2Grad) {} 2214 void set_align_corners(const bool &align_corners); 2215 bool get_align_corners() const; 2216 void set_half_pixel_centers(const bool &half_pixel_centers); 2217 bool get_half_pixel_centers() const; 2218 }; 2219 2220 class MIND_API ResizeNearestNeighborV2 : public BaseOperator { 2221 public: 2222 MIND_API_BASE_MEMBER(ResizeNearestNeighborV2); ResizeNearestNeighborV2()2223 ResizeNearestNeighborV2() : BaseOperator(kNameResizeNearestNeighborV2) {} 2224 void set_align_corners(const bool &align_corners); 2225 bool get_align_corners() const; 2226 void set_half_pixel_centers(const bool &half_pixel_centers); 2227 bool get_half_pixel_centers() const; 2228 }; 2229 2230 class MIND_API ReverseV2 : public BaseOperator { 2231 public: 2232 MIND_API_BASE_MEMBER(ReverseV2); ReverseV2()2233 ReverseV2() : BaseOperator(kNameReverseV2) {} 2234 void set_axis(const std::vector<int64_t> &axis); 2235 std::vector<int64_t> get_axis() const; 2236 }; 2237 2238 class MIND_API RFFTGrad : public BaseOperator { 2239 public: 2240 MIND_API_BASE_MEMBER(RFFTGrad); RFFTGrad()2241 RFFTGrad() : BaseOperator(kNameRFFTGrad) {} 2242 }; 2243 2244 class MIND_API RFFT : public BaseOperator { 2245 public: 2246 MIND_API_BASE_MEMBER(RFFT); RFFT()2247 RFFT() : BaseOperator(kNameRFFT) {} 2248 }; 2249 2250 class MIND_API RightShift : public BaseOperator { 2251 public: 2252 MIND_API_BASE_MEMBER(RightShift); RightShift()2253 RightShift() : BaseOperator(kNameRightShift) {} 2254 }; 2255 2256 class MIND_API RmsNormGrad : public BaseOperator { 2257 public: 2258 MIND_API_BASE_MEMBER(RmsNormGrad); RmsNormGrad()2259 RmsNormGrad() : BaseOperator(kNameRmsNormGrad) {} 2260 }; 2261 2262 class MIND_API RmsNorm : public BaseOperator { 2263 public: 2264 MIND_API_BASE_MEMBER(RmsNorm); RmsNorm()2265 RmsNorm() : BaseOperator(kNameRmsNorm) {} 2266 void set_epsilon(const float &epsilon); 2267 float get_epsilon() const; 2268 }; 2269 2270 class MIND_API Roll : public BaseOperator { 2271 public: 2272 MIND_API_BASE_MEMBER(Roll); Roll()2273 Roll() : BaseOperator(kNameRoll) {} 2274 void set_shift(const std::vector<int64_t> &shift); 2275 std::vector<int64_t> get_shift() const; 2276 void set_axis(const std::vector<int64_t> &axis); 2277 std::vector<int64_t> get_axis() const; 2278 }; 2279 2280 class MIND_API Round : public BaseOperator { 2281 public: 2282 MIND_API_BASE_MEMBER(Round); Round()2283 Round() : BaseOperator(kNameRound) {} 2284 }; 2285 2286 class MIND_API RsqrtGrad : public BaseOperator { 2287 public: 2288 MIND_API_BASE_MEMBER(RsqrtGrad); RsqrtGrad()2289 RsqrtGrad() : BaseOperator(kNameRsqrtGrad) {} 2290 }; 2291 2292 class MIND_API Rsqrt : public BaseOperator { 2293 public: 2294 MIND_API_BASE_MEMBER(Rsqrt); Rsqrt()2295 Rsqrt() : BaseOperator(kNameRsqrt) {} 2296 }; 2297 2298 class MIND_API ScalarAdd : public BaseOperator { 2299 public: 2300 MIND_API_BASE_MEMBER(ScalarAdd); ScalarAdd()2301 ScalarAdd() : BaseOperator(kNameScalarAdd) {} 2302 }; 2303 2304 class MIND_API ScalarBool : public BaseOperator { 2305 public: 2306 MIND_API_BASE_MEMBER(ScalarBool); ScalarBool()2307 ScalarBool() : BaseOperator(kNameScalarBool) {} 2308 }; 2309 2310 class MIND_API ScalarCast : public BaseOperator { 2311 public: 2312 MIND_API_BASE_MEMBER(ScalarCast); ScalarCast()2313 ScalarCast() : BaseOperator(kNameScalarCast) {} 2314 }; 2315 2316 class MIND_API ScalarDiv : public BaseOperator { 2317 public: 2318 MIND_API_BASE_MEMBER(ScalarDiv); ScalarDiv()2319 ScalarDiv() : BaseOperator(kNameScalarDiv) {} 2320 }; 2321 2322 class MIND_API ScalarEq : public BaseOperator { 2323 public: 2324 MIND_API_BASE_MEMBER(ScalarEq); ScalarEq()2325 ScalarEq() : BaseOperator(kNameScalarEq) {} 2326 }; 2327 2328 class MIND_API ScalarFloorDiv : public BaseOperator { 2329 public: 2330 MIND_API_BASE_MEMBER(ScalarFloorDiv); ScalarFloorDiv()2331 ScalarFloorDiv() : BaseOperator(kNameScalarFloorDiv) {} 2332 }; 2333 2334 class MIND_API ScalarGe : public BaseOperator { 2335 public: 2336 MIND_API_BASE_MEMBER(ScalarGe); ScalarGe()2337 ScalarGe() : BaseOperator(kNameScalarGe) {} 2338 }; 2339 2340 class MIND_API ScalarGt : public BaseOperator { 2341 public: 2342 MIND_API_BASE_MEMBER(ScalarGt); ScalarGt()2343 ScalarGt() : BaseOperator(kNameScalarGt) {} 2344 }; 2345 2346 class MIND_API ScalarLe : public BaseOperator { 2347 public: 2348 MIND_API_BASE_MEMBER(ScalarLe); ScalarLe()2349 ScalarLe() : BaseOperator(kNameScalarLe) {} 2350 }; 2351 2352 class MIND_API ScalarLog : public BaseOperator { 2353 public: 2354 MIND_API_BASE_MEMBER(ScalarLog); ScalarLog()2355 ScalarLog() : BaseOperator(kNameScalarLog) {} 2356 }; 2357 2358 class MIND_API ScalarLt : public BaseOperator { 2359 public: 2360 MIND_API_BASE_MEMBER(ScalarLt); ScalarLt()2361 ScalarLt() : BaseOperator(kNameScalarLt) {} 2362 }; 2363 2364 class MIND_API ScalarMod : public BaseOperator { 2365 public: 2366 MIND_API_BASE_MEMBER(ScalarMod); ScalarMod()2367 ScalarMod() : BaseOperator(kNameScalarMod) {} 2368 }; 2369 2370 class MIND_API ScalarMul : public BaseOperator { 2371 public: 2372 MIND_API_BASE_MEMBER(ScalarMul); ScalarMul()2373 ScalarMul() : BaseOperator(kNameScalarMul) {} 2374 }; 2375 2376 class MIND_API ScalarPow : public BaseOperator { 2377 public: 2378 MIND_API_BASE_MEMBER(ScalarPow); ScalarPow()2379 ScalarPow() : BaseOperator(kNameScalarPow) {} 2380 }; 2381 2382 class MIND_API ScalarSub : public BaseOperator { 2383 public: 2384 MIND_API_BASE_MEMBER(ScalarSub); ScalarSub()2385 ScalarSub() : BaseOperator(kNameScalarSub) {} 2386 }; 2387 2388 class MIND_API ScalarToTensor : public BaseOperator { 2389 public: 2390 MIND_API_BASE_MEMBER(ScalarToTensor); ScalarToTensor()2391 ScalarToTensor() : BaseOperator(kNameScalarToTensor) {} 2392 }; 2393 2394 class MIND_API ScalarUadd : public BaseOperator { 2395 public: 2396 MIND_API_BASE_MEMBER(ScalarUadd); ScalarUadd()2397 ScalarUadd() : BaseOperator(kNameScalarUadd) {} 2398 }; 2399 2400 class MIND_API ScalarUsub : public BaseOperator { 2401 public: 2402 MIND_API_BASE_MEMBER(ScalarUsub); ScalarUsub()2403 ScalarUsub() : BaseOperator(kNameScalarUsub) {} 2404 }; 2405 2406 class MIND_API ScatterAddExt : public BaseOperator { 2407 public: 2408 MIND_API_BASE_MEMBER(ScatterAddExt); ScatterAddExt()2409 ScatterAddExt() : BaseOperator(kNameScatterAddExt) {} 2410 }; 2411 2412 class MIND_API ScatterNd : public BaseOperator { 2413 public: 2414 MIND_API_BASE_MEMBER(ScatterNd); ScatterNd()2415 ScatterNd() : BaseOperator(kNameScatterNd) {} 2416 }; 2417 2418 class MIND_API Scatter : public BaseOperator { 2419 public: 2420 MIND_API_BASE_MEMBER(Scatter); Scatter()2421 Scatter() : BaseOperator(kNameScatter) {} 2422 }; 2423 2424 class MIND_API SearchSorted : public BaseOperator { 2425 public: 2426 MIND_API_BASE_MEMBER(SearchSorted); SearchSorted()2427 SearchSorted() : BaseOperator(kNameSearchSorted) {} 2428 void set_dtype(const int64_t &dtype); 2429 int64_t get_dtype() const; 2430 void set_right(const bool &right); 2431 bool get_right() const; 2432 }; 2433 2434 class MIND_API Select : public BaseOperator { 2435 public: 2436 MIND_API_BASE_MEMBER(Select); Select()2437 Select() : BaseOperator(kNameSelect) {} 2438 }; 2439 2440 class MIND_API SequenceConcat : public BaseOperator { 2441 public: 2442 MIND_API_BASE_MEMBER(SequenceConcat); SequenceConcat()2443 SequenceConcat() : BaseOperator(kNameSequenceConcat) {} 2444 void set_axis(const int64_t &axis); 2445 int64_t get_axis() const; 2446 }; 2447 2448 class MIND_API Shape : public BaseOperator { 2449 public: 2450 MIND_API_BASE_MEMBER(Shape); Shape()2451 Shape() : BaseOperator(kNameShape) {} 2452 }; 2453 2454 class MIND_API SigmoidGrad : public BaseOperator { 2455 public: 2456 MIND_API_BASE_MEMBER(SigmoidGrad); SigmoidGrad()2457 SigmoidGrad() : BaseOperator(kNameSigmoidGrad) {} 2458 }; 2459 2460 class MIND_API Sigmoid : public BaseOperator { 2461 public: 2462 MIND_API_BASE_MEMBER(Sigmoid); Sigmoid()2463 Sigmoid() : BaseOperator(kNameSigmoid) {} 2464 }; 2465 2466 class MIND_API Sign : public BaseOperator { 2467 public: 2468 MIND_API_BASE_MEMBER(Sign); Sign()2469 Sign() : BaseOperator(kNameSign) {} 2470 }; 2471 2472 class MIND_API SiLUGrad : public BaseOperator { 2473 public: 2474 MIND_API_BASE_MEMBER(SiLUGrad); SiLUGrad()2475 SiLUGrad() : BaseOperator(kNameSiLUGrad) {} 2476 }; 2477 2478 class MIND_API SiLU : public BaseOperator { 2479 public: 2480 MIND_API_BASE_MEMBER(SiLU); SiLU()2481 SiLU() : BaseOperator(kNameSiLU) {} 2482 }; 2483 2484 class MIND_API Sin : public BaseOperator { 2485 public: 2486 MIND_API_BASE_MEMBER(Sin); Sin()2487 Sin() : BaseOperator(kNameSin) {} 2488 }; 2489 2490 class MIND_API Sinc : public BaseOperator { 2491 public: 2492 MIND_API_BASE_MEMBER(Sinc); Sinc()2493 Sinc() : BaseOperator(kNameSinc) {} 2494 }; 2495 2496 class MIND_API Sinh : public BaseOperator { 2497 public: 2498 MIND_API_BASE_MEMBER(Sinh); Sinh()2499 Sinh() : BaseOperator(kNameSinh) {} 2500 }; 2501 2502 class MIND_API SliceExt : public BaseOperator { 2503 public: 2504 MIND_API_BASE_MEMBER(SliceExt); SliceExt()2505 SliceExt() : BaseOperator(kNameSliceExt) {} 2506 }; 2507 2508 class MIND_API SoftmaxBackward : public BaseOperator { 2509 public: 2510 MIND_API_BASE_MEMBER(SoftmaxBackward); SoftmaxBackward()2511 SoftmaxBackward() : BaseOperator(kNameSoftmaxBackward) {} 2512 }; 2513 2514 class MIND_API Softmax : public BaseOperator { 2515 public: 2516 MIND_API_BASE_MEMBER(Softmax); Softmax()2517 Softmax() : BaseOperator(kNameSoftmax) {} 2518 void set_axis(const std::vector<int64_t> &axis); 2519 std::vector<int64_t> get_axis() const; 2520 }; 2521 2522 class MIND_API SoftplusExt : public BaseOperator { 2523 public: 2524 MIND_API_BASE_MEMBER(SoftplusExt); SoftplusExt()2525 SoftplusExt() : BaseOperator(kNameSoftplusExt) {} 2526 }; 2527 2528 class MIND_API SoftplusGradExt : public BaseOperator { 2529 public: 2530 MIND_API_BASE_MEMBER(SoftplusGradExt); SoftplusGradExt()2531 SoftplusGradExt() : BaseOperator(kNameSoftplusGradExt) {} 2532 }; 2533 2534 class MIND_API SolveTriangular : public BaseOperator { 2535 public: 2536 MIND_API_BASE_MEMBER(SolveTriangular); SolveTriangular()2537 SolveTriangular() : BaseOperator(kNameSolveTriangular) {} 2538 }; 2539 2540 class MIND_API SortExt : public BaseOperator { 2541 public: 2542 MIND_API_BASE_MEMBER(SortExt); SortExt()2543 SortExt() : BaseOperator(kNameSortExt) {} 2544 }; 2545 2546 class MIND_API Split : public BaseOperator { 2547 public: 2548 MIND_API_BASE_MEMBER(Split); Split()2549 Split() : BaseOperator(kNameSplit) {} 2550 void set_axis(const int64_t &axis); 2551 int64_t get_axis() const; 2552 void set_output_num(const int64_t &output_num); 2553 int64_t get_output_num() const; 2554 }; 2555 2556 class MIND_API SplitTensor : public BaseOperator { 2557 public: 2558 MIND_API_BASE_MEMBER(SplitTensor); SplitTensor()2559 SplitTensor() : BaseOperator(kNameSplitTensor) {} 2560 }; 2561 2562 class MIND_API SplitWithSize : public BaseOperator { 2563 public: 2564 MIND_API_BASE_MEMBER(SplitWithSize); SplitWithSize()2565 SplitWithSize() : BaseOperator(kNameSplitWithSize) {} 2566 }; 2567 2568 class MIND_API SqrtGrad : public BaseOperator { 2569 public: 2570 MIND_API_BASE_MEMBER(SqrtGrad); SqrtGrad()2571 SqrtGrad() : BaseOperator(kNameSqrtGrad) {} 2572 }; 2573 2574 class MIND_API Sqrt : public BaseOperator { 2575 public: 2576 MIND_API_BASE_MEMBER(Sqrt); Sqrt()2577 Sqrt() : BaseOperator(kNameSqrt) {} 2578 }; 2579 2580 class MIND_API Square : public BaseOperator { 2581 public: 2582 MIND_API_BASE_MEMBER(Square); Square()2583 Square() : BaseOperator(kNameSquare) {} 2584 }; 2585 2586 class MIND_API StackExt : public BaseOperator { 2587 public: 2588 MIND_API_BASE_MEMBER(StackExt); StackExt()2589 StackExt() : BaseOperator(kNameStackExt) {} 2590 void set_dim(const int64_t &dim); 2591 int64_t get_dim() const; 2592 }; 2593 2594 class MIND_API StridedSlice : public BaseOperator { 2595 public: 2596 MIND_API_BASE_MEMBER(StridedSlice); StridedSlice()2597 StridedSlice() : BaseOperator(kNameStridedSlice) {} 2598 void set_begin_mask(const int64_t &begin_mask); 2599 int64_t get_begin_mask() const; 2600 void set_end_mask(const int64_t &end_mask); 2601 int64_t get_end_mask() const; 2602 void set_ellipsis_mask(const int64_t &ellipsis_mask); 2603 int64_t get_ellipsis_mask() const; 2604 void set_new_axis_mask(const int64_t &new_axis_mask); 2605 int64_t get_new_axis_mask() const; 2606 void set_shrink_axis_mask(const int64_t &shrink_axis_mask); 2607 int64_t get_shrink_axis_mask() const; 2608 }; 2609 2610 class MIND_API SubExt : public BaseOperator { 2611 public: 2612 MIND_API_BASE_MEMBER(SubExt); SubExt()2613 SubExt() : BaseOperator(kNameSubExt) {} 2614 }; 2615 2616 class MIND_API Sub : public BaseOperator { 2617 public: 2618 MIND_API_BASE_MEMBER(Sub); Sub()2619 Sub() : BaseOperator(kNameSub) {} 2620 }; 2621 2622 class MIND_API SumExt : public BaseOperator { 2623 public: 2624 MIND_API_BASE_MEMBER(SumExt); SumExt()2625 SumExt() : BaseOperator(kNameSumExt) {} 2626 }; 2627 2628 class MIND_API TanhGrad : public BaseOperator { 2629 public: 2630 MIND_API_BASE_MEMBER(TanhGrad); TanhGrad()2631 TanhGrad() : BaseOperator(kNameTanhGrad) {} 2632 }; 2633 2634 class MIND_API Tanh : public BaseOperator { 2635 public: 2636 MIND_API_BASE_MEMBER(Tanh); Tanh()2637 Tanh() : BaseOperator(kNameTanh) {} 2638 }; 2639 2640 class MIND_API TensorCopySlices : public BaseOperator { 2641 public: 2642 MIND_API_BASE_MEMBER(TensorCopySlices); TensorCopySlices()2643 TensorCopySlices() : BaseOperator(kNameTensorCopySlices) {} 2644 }; 2645 2646 class MIND_API TensorShape : public BaseOperator { 2647 public: 2648 MIND_API_BASE_MEMBER(TensorShape); TensorShape()2649 TensorShape() : BaseOperator(kNameTensorShape) {} 2650 }; 2651 2652 class MIND_API Tile : public BaseOperator { 2653 public: 2654 MIND_API_BASE_MEMBER(Tile); Tile()2655 Tile() : BaseOperator(kNameTile) {} 2656 }; 2657 2658 class MIND_API TopkExt : public BaseOperator { 2659 public: 2660 MIND_API_BASE_MEMBER(TopkExt); TopkExt()2661 TopkExt() : BaseOperator(kNameTopkExt) {} 2662 }; 2663 2664 class MIND_API TopKRouter : public BaseOperator { 2665 public: 2666 MIND_API_BASE_MEMBER(TopKRouter); TopKRouter()2667 TopKRouter() : BaseOperator(kNameTopKRouter) {} 2668 }; 2669 2670 class MIND_API Trace : public BaseOperator { 2671 public: 2672 MIND_API_BASE_MEMBER(Trace); Trace()2673 Trace() : BaseOperator(kNameTrace) {} 2674 }; 2675 2676 class MIND_API Transpose : public BaseOperator { 2677 public: 2678 MIND_API_BASE_MEMBER(Transpose); Transpose()2679 Transpose() : BaseOperator(kNameTranspose) {} 2680 }; 2681 2682 class MIND_API Triu : public BaseOperator { 2683 public: 2684 MIND_API_BASE_MEMBER(Triu); Triu()2685 Triu() : BaseOperator(kNameTriu) {} 2686 void set_diagonal(const int64_t &diagonal); 2687 int64_t get_diagonal() const; 2688 }; 2689 2690 class MIND_API TupleToList : public BaseOperator { 2691 public: 2692 MIND_API_BASE_MEMBER(TupleToList); TupleToList()2693 TupleToList() : BaseOperator(kNameTupleToList) {} 2694 }; 2695 2696 class MIND_API TupleToTensor : public BaseOperator { 2697 public: 2698 MIND_API_BASE_MEMBER(TupleToTensor); TupleToTensor()2699 TupleToTensor() : BaseOperator(kNameTupleToTensor) {} 2700 }; 2701 2702 class MIND_API UniformExt : public BaseOperator { 2703 public: 2704 MIND_API_BASE_MEMBER(UniformExt); UniformExt()2705 UniformExt() : BaseOperator(kNameUniformExt) {} 2706 }; 2707 2708 class MIND_API Unique2 : public BaseOperator { 2709 public: 2710 MIND_API_BASE_MEMBER(Unique2); Unique2()2711 Unique2() : BaseOperator(kNameUnique2) {} 2712 }; 2713 2714 class MIND_API UniqueDim : public BaseOperator { 2715 public: 2716 MIND_API_BASE_MEMBER(UniqueDim); UniqueDim()2717 UniqueDim() : BaseOperator(kNameUniqueDim) {} 2718 }; 2719 2720 class MIND_API UnsortedSegmentSum : public BaseOperator { 2721 public: 2722 MIND_API_BASE_MEMBER(UnsortedSegmentSum); UnsortedSegmentSum()2723 UnsortedSegmentSum() : BaseOperator(kNameUnsortedSegmentSum) {} 2724 }; 2725 2726 class MIND_API UnstackExt : public BaseOperator { 2727 public: 2728 MIND_API_BASE_MEMBER(UnstackExt); UnstackExt()2729 UnstackExt() : BaseOperator(kNameUnstackExt) {} 2730 void set_axis(const int64_t &axis); 2731 int64_t get_axis() const; 2732 }; 2733 2734 class MIND_API UpsampleBilinear2DGrad : public BaseOperator { 2735 public: 2736 MIND_API_BASE_MEMBER(UpsampleBilinear2DGrad); UpsampleBilinear2DGrad()2737 UpsampleBilinear2DGrad() : BaseOperator(kNameUpsampleBilinear2DGrad) {} 2738 }; 2739 2740 class MIND_API UpsampleBilinear2D : public BaseOperator { 2741 public: 2742 MIND_API_BASE_MEMBER(UpsampleBilinear2D); UpsampleBilinear2D()2743 UpsampleBilinear2D() : BaseOperator(kNameUpsampleBilinear2D) {} 2744 }; 2745 2746 class MIND_API UpsampleLinear1DGrad : public BaseOperator { 2747 public: 2748 MIND_API_BASE_MEMBER(UpsampleLinear1DGrad); UpsampleLinear1DGrad()2749 UpsampleLinear1DGrad() : BaseOperator(kNameUpsampleLinear1DGrad) {} 2750 }; 2751 2752 class MIND_API UpsampleLinear1D : public BaseOperator { 2753 public: 2754 MIND_API_BASE_MEMBER(UpsampleLinear1D); UpsampleLinear1D()2755 UpsampleLinear1D() : BaseOperator(kNameUpsampleLinear1D) {} 2756 }; 2757 2758 class MIND_API UpsampleNearest1DGrad : public BaseOperator { 2759 public: 2760 MIND_API_BASE_MEMBER(UpsampleNearest1DGrad); UpsampleNearest1DGrad()2761 UpsampleNearest1DGrad() : BaseOperator(kNameUpsampleNearest1DGrad) {} 2762 }; 2763 2764 class MIND_API UpsampleNearest1D : public BaseOperator { 2765 public: 2766 MIND_API_BASE_MEMBER(UpsampleNearest1D); UpsampleNearest1D()2767 UpsampleNearest1D() : BaseOperator(kNameUpsampleNearest1D) {} 2768 }; 2769 2770 class MIND_API UpsampleNearest2DGrad : public BaseOperator { 2771 public: 2772 MIND_API_BASE_MEMBER(UpsampleNearest2DGrad); UpsampleNearest2DGrad()2773 UpsampleNearest2DGrad() : BaseOperator(kNameUpsampleNearest2DGrad) {} 2774 }; 2775 2776 class MIND_API UpsampleNearest2D : public BaseOperator { 2777 public: 2778 MIND_API_BASE_MEMBER(UpsampleNearest2D); UpsampleNearest2D()2779 UpsampleNearest2D() : BaseOperator(kNameUpsampleNearest2D) {} 2780 }; 2781 2782 class MIND_API UpsampleNearest3DGrad : public BaseOperator { 2783 public: 2784 MIND_API_BASE_MEMBER(UpsampleNearest3DGrad); UpsampleNearest3DGrad()2785 UpsampleNearest3DGrad() : BaseOperator(kNameUpsampleNearest3DGrad) {} 2786 }; 2787 2788 class MIND_API UpsampleNearest3D : public BaseOperator { 2789 public: 2790 MIND_API_BASE_MEMBER(UpsampleNearest3D); UpsampleNearest3D()2791 UpsampleNearest3D() : BaseOperator(kNameUpsampleNearest3D) {} 2792 }; 2793 2794 class MIND_API UpsampleTrilinear3DGrad : public BaseOperator { 2795 public: 2796 MIND_API_BASE_MEMBER(UpsampleTrilinear3DGrad); UpsampleTrilinear3DGrad()2797 UpsampleTrilinear3DGrad() : BaseOperator(kNameUpsampleTrilinear3DGrad) {} 2798 void set_align_corners(const bool &align_corners); 2799 bool get_align_corners() const; 2800 }; 2801 2802 class MIND_API UpsampleTrilinear3D : public BaseOperator { 2803 public: 2804 MIND_API_BASE_MEMBER(UpsampleTrilinear3D); UpsampleTrilinear3D()2805 UpsampleTrilinear3D() : BaseOperator(kNameUpsampleTrilinear3D) {} 2806 void set_align_corners(const bool &align_corners); 2807 bool get_align_corners() const; 2808 }; 2809 2810 class MIND_API View : public BaseOperator { 2811 public: 2812 MIND_API_BASE_MEMBER(View); View()2813 View() : BaseOperator(kNameView) {} 2814 }; 2815 2816 class MIND_API ZerosLikeExt : public BaseOperator { 2817 public: 2818 MIND_API_BASE_MEMBER(ZerosLikeExt); ZerosLikeExt()2819 ZerosLikeExt() : BaseOperator(kNameZerosLikeExt) {} 2820 }; 2821 2822 class MIND_API ZerosLike : public BaseOperator { 2823 public: 2824 MIND_API_BASE_MEMBER(ZerosLike); ZerosLike()2825 ZerosLike() : BaseOperator(kNameZerosLike) {} 2826 }; 2827 2828 class MIND_API Zeros : public BaseOperator { 2829 public: 2830 MIND_API_BASE_MEMBER(Zeros); Zeros()2831 Zeros() : BaseOperator(kNameZeros) {} 2832 }; 2833 2834 class MIND_API DynamicQuantExt : public BaseOperator { 2835 public: 2836 MIND_API_BASE_MEMBER(DynamicQuantExt); DynamicQuantExt()2837 DynamicQuantExt() : BaseOperator(kNameDynamicQuantExt) {} 2838 }; 2839 2840 class MIND_API FusedInferAttentionScore : public BaseOperator { 2841 public: 2842 MIND_API_BASE_MEMBER(FusedInferAttentionScore); FusedInferAttentionScore()2843 FusedInferAttentionScore() : BaseOperator(kNameFusedInferAttentionScore) {} 2844 void set_num_heads(const int64_t &num_heads); 2845 int64_t get_num_heads() const; 2846 void set_scale_value(const float &scale_value); 2847 float get_scale_value() const; 2848 void set_pre_tokens(const int64_t &pre_tokens); 2849 int64_t get_pre_tokens() const; 2850 void set_next_tokens(const int64_t &next_tokens); 2851 int64_t get_next_tokens() const; 2852 void set_input_layout(const int64_t &input_layout); 2853 int64_t get_input_layout() const; 2854 void set_num_key_value_heads(const int64_t &num_key_value_heads); 2855 int64_t get_num_key_value_heads() const; 2856 void set_sparse_mode(const int64_t &sparse_mode); 2857 int64_t get_sparse_mode() const; 2858 void set_inner_precise(const int64_t &inner_precise); 2859 int64_t get_inner_precise() const; 2860 void set_block_size(const int64_t &block_size); 2861 int64_t get_block_size() const; 2862 void set_antiquant_mode(const int64_t &antiquant_mode); 2863 int64_t get_antiquant_mode() const; 2864 void set_softmax_lse_flag(const bool &softmax_lse_flag); 2865 bool get_softmax_lse_flag() const; 2866 }; 2867 2868 class MIND_API GroupedMatmul : public BaseOperator { 2869 public: 2870 MIND_API_BASE_MEMBER(GroupedMatmul); GroupedMatmul()2871 GroupedMatmul() : BaseOperator(kNameGroupedMatmul) {} 2872 void set_split_item(const int64_t &split_item); 2873 int64_t get_split_item() const; 2874 void set_group_type(const int64_t &group_type); 2875 int64_t get_group_type() const; 2876 }; 2877 2878 class MIND_API KVCacheScatterUpdate : public BaseOperator { 2879 public: 2880 MIND_API_BASE_MEMBER(KVCacheScatterUpdate); KVCacheScatterUpdate()2881 KVCacheScatterUpdate() : BaseOperator(kNameKVCacheScatterUpdate) {} 2882 }; 2883 2884 class MIND_API MatmulBiasSplitOut2 : public BaseOperator { 2885 public: 2886 MIND_API_BASE_MEMBER(MatmulBiasSplitOut2); MatmulBiasSplitOut2()2887 MatmulBiasSplitOut2() : BaseOperator(kNameMatmulBiasSplitOut2) {} 2888 }; 2889 2890 class MIND_API MatmulBiasSplitOut3 : public BaseOperator { 2891 public: 2892 MIND_API_BASE_MEMBER(MatmulBiasSplitOut3); MatmulBiasSplitOut3()2893 MatmulBiasSplitOut3() : BaseOperator(kNameMatmulBiasSplitOut3) {} 2894 }; 2895 2896 class MIND_API MatmulBiasSplitSiluOut2 : public BaseOperator { 2897 public: 2898 MIND_API_BASE_MEMBER(MatmulBiasSplitSiluOut2); MatmulBiasSplitSiluOut2()2899 MatmulBiasSplitSiluOut2() : BaseOperator(kNameMatmulBiasSplitSiluOut2) {} 2900 }; 2901 2902 class MIND_API MatmulSplitOut2 : public BaseOperator { 2903 public: 2904 MIND_API_BASE_MEMBER(MatmulSplitOut2); MatmulSplitOut2()2905 MatmulSplitOut2() : BaseOperator(kNameMatmulSplitOut2) {} 2906 }; 2907 2908 class MIND_API MatmulSplitOut3 : public BaseOperator { 2909 public: 2910 MIND_API_BASE_MEMBER(MatmulSplitOut3); MatmulSplitOut3()2911 MatmulSplitOut3() : BaseOperator(kNameMatmulSplitOut3) {} 2912 }; 2913 2914 class MIND_API MatmulSplitSiluOut2 : public BaseOperator { 2915 public: 2916 MIND_API_BASE_MEMBER(MatmulSplitSiluOut2); MatmulSplitSiluOut2()2917 MatmulSplitSiluOut2() : BaseOperator(kNameMatmulSplitSiluOut2) {} 2918 }; 2919 2920 class MIND_API MoeFinalizeRouting : public BaseOperator { 2921 public: 2922 MIND_API_BASE_MEMBER(MoeFinalizeRouting); MoeFinalizeRouting()2923 MoeFinalizeRouting() : BaseOperator(kNameMoeFinalizeRouting) {} 2924 }; 2925 2926 class MIND_API QuantBatchMatmul : public BaseOperator { 2927 public: 2928 MIND_API_BASE_MEMBER(QuantBatchMatmul); QuantBatchMatmul()2929 QuantBatchMatmul() : BaseOperator(kNameQuantBatchMatmul) {} 2930 void set_transpose_x1(const bool &transpose_x1); 2931 bool get_transpose_x1() const; 2932 void set_transpose_x2(const bool &transpose_x2); 2933 bool get_transpose_x2() const; 2934 void set_dtype(const int64_t &dtype); 2935 int64_t get_dtype() const; 2936 }; 2937 2938 class MIND_API QuantV2 : public BaseOperator { 2939 public: 2940 MIND_API_BASE_MEMBER(QuantV2); QuantV2()2941 QuantV2() : BaseOperator(kNameQuantV2) {} 2942 }; 2943 2944 class MIND_API QuantbatchmatmulSplitOut2 : public BaseOperator { 2945 public: 2946 MIND_API_BASE_MEMBER(QuantbatchmatmulSplitOut2); QuantbatchmatmulSplitOut2()2947 QuantbatchmatmulSplitOut2() : BaseOperator(kNameQuantbatchmatmulSplitOut2) {} 2948 }; 2949 2950 class MIND_API QuantbatchmatmulSplitOut3 : public BaseOperator { 2951 public: 2952 MIND_API_BASE_MEMBER(QuantbatchmatmulSplitOut3); QuantbatchmatmulSplitOut3()2953 QuantbatchmatmulSplitOut3() : BaseOperator(kNameQuantbatchmatmulSplitOut3) {} 2954 }; 2955 2956 class MIND_API QuantbatchmatmulSplitSiluOut2 : public BaseOperator { 2957 public: 2958 MIND_API_BASE_MEMBER(QuantbatchmatmulSplitSiluOut2); QuantbatchmatmulSplitSiluOut2()2959 QuantbatchmatmulSplitSiluOut2() : BaseOperator(kNameQuantbatchmatmulSplitSiluOut2) {} 2960 }; 2961 2962 class MIND_API WeightQuantBatchMatmul : public BaseOperator { 2963 public: 2964 MIND_API_BASE_MEMBER(WeightQuantBatchMatmul); WeightQuantBatchMatmul()2965 WeightQuantBatchMatmul() : BaseOperator(kNameWeightQuantBatchMatmul) {} 2966 void set_transpose_x(const bool &transpose_x); 2967 bool get_transpose_x() const; 2968 void set_transpose_weight(const bool &transpose_weight); 2969 bool get_transpose_weight() const; 2970 void set_antiquant_group_size(const int64_t &antiquant_group_size); 2971 int64_t get_antiquant_group_size() const; 2972 }; 2973 2974 } // namespace mindspore::ops 2975 #endif // MINDSPORE_CORE_OPS_GEN_LITE_OPS_H_ 2976