ops.yaml

# This file is designed for C++ operators, which manages the
# generated code for dynamic mode and static mode. If you want
# to add the new operator configuration, make sure an operator's
# Python API, dynamic graph API, and static graph Operator parameters
# are consistent and correspond one-to-one. It's forbidden that the
# operator configured in this yaml file does not have Python API.

  # interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : abs
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : RealAndImagInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : abs
    data_type : x
  inplace: (x -> out)
  backward : abs_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : accuracy
  args : (Tensor x, Tensor indices, Tensor label)
  output : Tensor(accuracy), Tensor(correct), Tensor(total)
  infer_meta :
    func : AccuracyInferMeta
  kernel :
    func : accuracy
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : accuracy_check
  args : (Tensor x, Tensor y, str fn_name, double rtol=1e-5, double atol=1e-8,  bool equal_nan=false)
  output : Tensor(out)
  infer_meta :
    func : ValueCompareInferMeta
    param: [x, y]
  kernel :
    func : accuracy_check
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : acos
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : acos
  inplace: (x -> out)
  backward : acos_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : acosh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : acosh
  inplace: (x -> out)
  backward : acosh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : adadelta_
  args : (Tensor param, Tensor grad, Tensor avg_squared_grad, Tensor avg_squared_update,
    Tensor learning_rate, Tensor master_param, float rho = 0.95f, float epsilon =
    1.0e-6f, bool multi_precision = false)
  output : Tensor(param_out), Tensor(moment_out), Tensor(inf_norm_out), Tensor(master_param_out)
  infer_meta :
    func : AdadeltaInferMeta
  kernel :
    func : adadelta
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (avg_squared_grad -> moment_out), (avg_squared_update -> inf_norm_out), (master_param -> master_param_out)
  traits : paddle::dialect::ForwardOnlyTrait

- op : adagrad_
  args : (Tensor param, Tensor grad, Tensor moment, Tensor learning_rate, Tensor master_param, float epsilon = 1.0e-6f, bool multi_precision = false)
  output : Tensor(param_out), Tensor(moment_out), Tensor(master_param_out)
  infer_meta :
    func : AdagradInferMeta
  kernel :
    func : adagrad {dense, dense, dense, dense, dense -> dense, dense, dense}
           adagrad_dense_param_sparse_grad {dense, selected_rows, dense, dense, dense -> dense, dense, dense}
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (moment -> moment_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : adam_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment1, Tensor moment2, Tensor moment2_max, Tensor beta1_pow, Tensor beta2_pow, Tensor master_param, Tensor skip_update, Scalar beta1 = 0.9f, Scalar beta2 = 0.999f, Scalar epsilon = 1.0e-8f, bool lazy_mode = false, int64_t min_row_size_to_use_multithread = 1000, bool multi_precision = false, bool use_global_beta_pow = false, bool amsgrad = false)
  output : Tensor(param_out), Tensor(moment1_out), Tensor(moment2_out), Tensor(moment2_max_out), Tensor(beta1_pow_out), Tensor(beta2_pow_out), Tensor(master_param_out)
  infer_meta :
    func : AdamInferMeta
    spmd_rule : AdamInferSpmdDynamic
  kernel :
    func : adam {dense, dense, dense, dense, dense, dense, dense, dense, dense, dense -> dense, dense, dense, dense, dense, dense, dense},
           adam_dense_param_sparse_grad {dense, selected_rows, dense, dense, dense, dense, dense, dense, dense, dense -> dense, dense, dense, dense, dense, dense, dense}
    data_type : param
  optional : moment2_max, master_param, skip_update, moment2_max_out, master_param_out
  inplace : (param -> param_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (moment2_max -> moment2_max_out), (beta1_pow -> beta1_pow_out), (beta2_pow -> beta2_pow_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : adamax_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment, Tensor inf_norm, Tensor beta1_pow, Tensor master_param, float beta1 = 0.9f, float beta2 = 0.999f, float epsilon = 1.0e-8f, bool multi_precision = false)
  output : Tensor(param_out), Tensor(moment_out), Tensor(inf_norm_out), Tensor(master_param_out)
  infer_meta :
    func : AdamaxInferMeta
  kernel :
    func : adamax
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (moment -> moment_out), (inf_norm -> inf_norm_out), (master_param ->master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : adamw_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment1, Tensor moment2, Tensor moment2_max, Tensor beta1_pow, Tensor beta2_pow, Tensor master_param, Tensor skip_update, Scalar beta1 = 0.9f, Scalar beta2 = 0.999f, Scalar epsilon = 1.0e-8f, float lr_ratio = 1.0f, float coeff = 0.01f, bool with_decay = false, bool lazy_mode = false, int64_t min_row_size_to_use_multithread = 1000, bool multi_precision = false, bool use_global_beta_pow = false, bool amsgrad = false)
  output : Tensor(param_out), Tensor(moment1_out), Tensor(moment2_out), Tensor(moment2_max_out), Tensor(beta1_pow_out), Tensor(beta2_pow_out), Tensor(master_param_out)
  infer_meta :
    func : AdamwInferMeta
    spmd_rule : AdamwInferSpmdDynamic
  kernel :
    func : adamw
    data_type : param
  optional : moment2_max, master_param, skip_update, moment2_max_out, master_param_out
  inplace : (param -> param_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (moment2_max -> moment2_max_out), (beta1_pow -> beta1_pow_out), (beta2_pow -> beta2_pow_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : add_position_encoding
  args: (Tensor x, float alpha = 1.0f, float beta = 1.0f)
  output: Tensor (out)
  infer_meta:
    func: AddPositionEncodingInferMeta
  kernel:
    func: add_position_encoding
    data_type: x
  backward: add_position_encoding_grad

- op : addmm
  args : (Tensor input, Tensor x, Tensor y, float beta=1.0, float alpha=1.0)
  output : Tensor(out)
  infer_meta :
    func : AddmmInferMeta
  kernel :
    func : addmm
    data_type : x
  inplace: (input -> out)
  backward : addmm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : affine_channel
  args: (Tensor x, Tensor scale, Tensor bias, str data_layout = "AnyLayout")
  output: Tensor (out)
  infer_meta:
    func: AffineChannelInferMeta
  kernel:
    func: affine_channel
  backward: affine_channel_grad
  inplace : (x -> out)

- op : affine_grid
  args : (Tensor input, IntArray output_shape={}, bool align_corners=true)
  output : Tensor(output)
  infer_meta :
    func : AffineGridInferMeta
    param : [input, output_shape, align_corners]
  kernel :
    func : affine_grid
    param : [input, output_shape, align_corners]
    data_type : input
  backward : affine_grid_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : all
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceInferMeta
    spmd_rule : ReductionAllInferSpmdDynamic
  kernel :
    func : all
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : all_gather
  args : (Tensor x, int ring_id = 0, int nranks=0)
  output : Tensor(out)
  infer_meta :
    func : AllGatherInferMeta
    param: [x, nranks]
  kernel :
    func : all_gather
    param: [x, nranks]
  traits : paddle::dialect::ForwardOnlyTrait

- op : all_reduce
  args : (Tensor x, int ring_id = 0, int reduce_type = 0)
  output : Tensor(out)
  infer_meta :
    func : AllReduceInferMeta
    param: [x]
  kernel :
    func : all_reduce
    param: [x, reduce_type]
  inplace : (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : all_to_all
  args : (Tensor x, int ring_id = 0)
  output : Tensor(out)
  infer_meta :
    func : AllToAllInferMeta
    param: [x]
  kernel :
    func : all_to_all
    param: [x]

- op : allclose
  args : (Tensor x, Tensor y, Scalar(double) rtol=1e-5, Scalar(double) atol=1e-8, bool equal_nan=false)
  output : Tensor(out)
  infer_meta :
    func : AllValueCompareInferMeta
    param: [x, y]
  kernel :
    func : allclose
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : amax
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceInferMeta
  kernel :
    func : amax
  backward : amax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : amin
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceInferMeta
  kernel :
    func : amin
  backward : amin_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : angle
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : RealAndImagInferMeta
  kernel :
    func : angle
  backward : angle_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : pir::UnaryElementWiseTrait

- op : any
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceInferMeta
    spmd_rule : ReductionAnyInferSpmdDynamic
  kernel :
    func : any
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : ap_facade
  args : (Tensor[] xs, int64_t num_outputs, str custom_op_name, str infer_meta_func_name, str infer_symbolic_func_name, str serialized_attributes)
  output : Tensor[](out){num_outputs}
  optional : xs
  infer_meta :
    func : ApFacadeInferMeta
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  kernel :
    func : ap_facade
  traits : paddle::dialect::ForwardOnlyTrait

- op : ap_trivial_fusion_begin
  args : (Tensor[] xs)
  output : Tensor(out)
  optional : xs
  infer_meta :
    func : ApTrivialFusionBeginInferMeta
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  kernel :
    func : ap_trivial_fusion_begin
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : ap_trivial_fusion_end
  args : (Tensor[] xs)
  output : Tensor(out)
  optional : xs
  infer_meta :
    func : ApTrivialFusionEndInferMeta
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  kernel :
    func : ap_trivial_fusion_end
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : ap_variadic
  args : (Tensor[] xs, int num_outputs, str code_module_lambda, str infer_symbolic_lambda, str infer_meta_lambda, str rnel_dispatch_lambda, str kernel_dispatch_const_data_lambda)
  output : Tensor[](out){num_outputs}
  infer_meta :
    func : ApVariadicInferMeta
  kernel :
    func : ap_variadic
  traits : paddle::dialect::ForwardOnlyTrait

- op : apply_per_channel_scale
  args: (Tensor x, Tensor scales)
  output: Tensor(out)
  infer_meta :
    func : ApplyPerChannelScaleInferMeta
  kernel :
    func : apply_per_channel_scale
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : argmax
  args : (Tensor x, Scalar(int64_t) axis, bool keepdims = false, bool flatten = false, DataType dtype = DataType::INT64)
  output : Tensor(out)
  infer_meta :
    func : ArgMinMaxInferMeta
    spmd_rule : ArgMaxInferSpmdDynamic
  kernel :
    func : argmax
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : argmin
  args : (Tensor x, Scalar(int64_t) axis, bool keepdims = false, bool flatten = false, DataType dtype = DataType::INT64)
  output : Tensor(out)
  infer_meta :
    func : ArgMinMaxInferMeta
    spmd_rule : ArgMinInferSpmdDynamic
  kernel :
    func : argmin
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : argsort
  args : (Tensor x, int axis=-1, bool descending=false, bool stable=false)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : ArgsortInferMeta
    spmd_rule : ArgSortInferSpmd
  kernel :
    func : argsort
  backward : argsort_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : as_complex
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : AsComplexInferMeta
  kernel :
    func : as_complex
  backward : as_complex_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : as_real
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : AsRealInferMeta
  kernel :
    func : as_real
  backward : as_real_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : as_strided
  args : (Tensor input, int64_t[] dims = {}, int64_t[] stride = {}, int64_t offset = 0)
  output : Tensor
  infer_meta :
    func : StridedUnChangedInferMeta
    param : [input]
  kernel :
    func : as_strided
  backward : as_strided_grad
  no_need_buffer : input
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : asgd_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor d, Tensor y, Tensor n, Tensor master_param, bool multi_precision=false)
  output : Tensor(param_out), Tensor(d_out), Tensor(y_out), Tensor(master_param_out)
  infer_meta :
    func : ASGDInferMeta
  kernel :
    func : asgd
    data_type : param
  data_transform :
    support_trans_dtype : learning_rate, n
  optional : master_param, master_param_out
  inplace : (param -> param_out), (d -> d_out), (y -> y_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : asin
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : asin
  inplace: (x -> out)
  backward : asin_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : asinh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : asinh
  inplace: (x -> out)
  backward : asinh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : assign_out_
  args : (Tensor x, Tensor output)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : assign
    param : [x]
  inplace : (output -> out)
  backward : assign_out__grad
  traits : pir::SideEffectTrait
  interfaces : paddle::dialect::LayoutTransformationInterface

- op : assign_pos
  args : (Tensor x, Tensor cum_count, Tensor eff_num_len)
  output : Tensor(out)
  infer_meta :
    func : AssignPosInferMeta
  kernel :
    func : assign_pos
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : assign_value_
  args : (Tensor output, int[] shape, DataType dtype, Scalar[] values, Place place = {})
  output : Tensor(out)
  inplace: (output -> out)
  infer_meta :
    func : AssignValueInferMeta
    param : [shape, dtype]
  kernel :
    func : assign_value
    param : [shape, dtype, values]
    data_type : dtype
    backend : place > output
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : atan
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : atan
  inplace: (x -> out)
  backward : atan_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : atan2
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : Atan2InferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : atan2
  backward : atan2_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : atanh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : atanh
  inplace: (x -> out)
  backward : atanh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : attention_lstm
  args: (Tensor x, Tensor c0, Tensor h0, Tensor attention_weight, Tensor attention_bias,
    Tensor attention_scalar, Tensor attention_scalar_bias, Tensor lstm_weight,
    Tensor lstm_bias, str gate_activation = "sigmoid", str cell_activation = "tanh",
    str candidate_activation = "tanh")
  output: Tensor (hidden), Tensor (cell), Tensor (attentioned_x), Tensor (attention_fc_out),
    Tensor (lstm_x), Tensor (lstm_out)
  infer_meta:
    func: AttentionLstmInferMeta
  kernel:
    func: attention_lstm
    data_type: x
  optional: h0, attention_bias, attention_scalar, attention_scalar_bias
  intermediate: attentioned_x, attention_fc_out, lstm_x, lstm_out
  traits : paddle::dialect::ForwardOnlyTrait

- op : auc
  args : (Tensor x, Tensor label, Tensor stat_pos, Tensor stat_neg, Tensor ins_tag_weight, str curve = "ROC", int num_thresholds = (2 << 12) - 1, int slide_steps = 1)
  output : Tensor(auc), Tensor(stat_pos_out), Tensor(stat_neg_out)
  infer_meta :
    func : AucInferMeta
  kernel :
    func : auc
    data_type : x
  optional : ins_tag_weight
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : average_accumulates_
  args : (Tensor param, Tensor in_sum_1, Tensor in_sum_2, Tensor in_sum_3, Tensor in_num_accumulates, Tensor in_old_num_accumulates, Tensor in_num_updates, float average_window = 0, int64_t max_average_window = INT64_MAX, int64_t min_average_window = 10000L)
  output : Tensor(out_sum_1), Tensor(out_sum_2), Tensor(out_sum_3), Tensor(out_num_accumulates), Tensor(out_old_num_accumulates), Tensor(out_num_updates)
  infer_meta:
    func : AverageAccumulatesInferMeta
  kernel :
    func : average_accumulates {dense, dense, dense, dense, dense ,dense, dense -> dense, dense, dense, dense, dense, dense}
    data_type : param
  inplace : (in_sum_1 -> out_sum_1), (in_sum_2 -> out_sum_2), (in_sum_3 -> out_sum_3), (in_num_accumulates -> out_num_accumulates), (in_old_num_accumulates -> out_old_num_accumulates), (in_num_updates -> out_num_updates)
  traits : paddle::dialect::ForwardOnlyTrait

- op : baddbmm
  args : (Tensor input, Tensor x, Tensor y, float beta=1.0, float alpha=1.0, DataType out_dtype=DataType::UNDEFINED)
  output : Tensor(out)
  infer_meta :
    func : BaddbmmInferMeta
  kernel :
    func : baddbmm
    data_type : x
  inplace: (input -> out)
  backward : baddbmm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : barrier
  args : (Tensor x, int ring_id=0)
  output : Tensor(out)
  infer_meta:
    func : BarrierInferMeta
    param: [x]
  kernel :
    func : barrier
    param: [x]

- op : batch_fc
  args : (Tensor input, Tensor w, Tensor bias)
  output : Tensor(out)
  infer_meta:
    func : BatchFCInferMeta
  kernel :
    func : batch_fc
    data_type: input
  backward: batch_fc_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : bce_loss
  args : (Tensor input, Tensor label)
  output : Tensor(out)
  infer_meta :
    func : BCELossInferMeta
  kernel :
    func : bce_loss
    data_type : input
  inplace : (input -> out)
  backward : bce_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : beam_search
  args: (Tensor pre_ids, Tensor pre_scores, Tensor ids, Tensor scores, int level,
    int beam_size, int end_id, bool is_accumulated = true)
  output: Tensor (selected_ids), Tensor (selected_scores), Tensor (parent_idx)
  infer_meta:
    func: BeamSearchInferMeta
  kernel:
    func: beam_search
    data_type: pre_ids
  optional: ids, parent_idx
  traits : paddle::dialect::ForwardOnlyTrait

- op : bernoulli
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : bernoulli
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : bicubic_interp
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : bicubic_interp
    data_type : x
  backward : bicubic_interp_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : bilinear
  args : (Tensor x, Tensor y, Tensor weight, Tensor bias)
  output : Tensor
  infer_meta :
    func : BilinearInferMeta
  kernel :
    func : bilinear
  optional : bias
  backward : bilinear_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : bilinear_interp
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : bilinear_interp
    data_type : x
  backward : bilinear_interp_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : bincount
  args: (Tensor x, Tensor weights, Scalar(int) minlength = 0)
  output: Tensor(out)
  infer_meta:
    func: BincountInferMeta
  kernel:
    func: bincount
  optional: weights
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : binomial
  args : (Tensor count, Tensor prob)
  output : Tensor(out)
  infer_meta :
    func : BinomialInferMeta
  kernel :
    func : binomial
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : bipartite_match
  args: (Tensor dist_mat, str match_type = "bipartite", float dist_threshold = 0.5)
  output: Tensor (col_to_row_match_indices), Tensor (col_to_row_match_dist)
  infer_meta:
    func: BipartiteMatchInferMeta
  kernel:
    func: bipartite_match
    data_type: dist_mat
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : bitwise_and
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : bitwise_and
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::BinaryElementWiseTrait

- op : bitwise_left_shift
  args : (Tensor x, Tensor y, bool is_arithmetic = true)
  output : Tensor(out)
  infer_meta :
    func : BitwiseShiftInferMeta
  kernel :
    func : bitwise_left_shift
    backend : x
  inplace: (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : bitwise_not
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : bitwise_not
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : bitwise_or
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : bitwise_or
    backend : x
  inplace: (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait, pir::BinaryElementWiseTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : bitwise_right_shift
  args : (Tensor x, Tensor y, bool is_arithmetic = true)
  output : Tensor(out)
  infer_meta :
    func : BitwiseShiftInferMeta
  kernel :
    func : bitwise_right_shift
    backend : x
  inplace: (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : bitwise_xor
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : bitwise_xor
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::BinaryElementWiseTrait

- op : bmm
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : BmmInferMeta
    spmd_rule: BmmInferSpmd
  kernel :
    func : bmm
  backward : bmm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : box_clip
  args: (Tensor input, Tensor im_info)
  output: Tensor (output)
  infer_meta:
    func: BoxClipInferMeta
  kernel:
    func: box_clip
  interfaces: paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : box_coder
  args : (Tensor prior_box, Tensor prior_box_var, Tensor target_box, str code_type = "encode_center_size", bool box_normalized = true, int axis = 0, float[] variance = {})
  output : Tensor(output_box)
  infer_meta :
    func : BoxCoderInferMeta
  kernel :
    func : box_coder
  optional : prior_box_var
  interfaces: paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : broadcast
  args : (Tensor x, int ring_id = 0, int root = 0)
  output : Tensor(out)
  infer_meta :
    func : DistBroadcastInferMeta
    param: [x]
  kernel :
    func : broadcast
    param: [x, root]
  inplace : (x -> out)

- op : broadcast_tensors
  args: (Tensor[] input)
  output: Tensor[]{input.size()}
  infer_meta:
    func: BroadcastTensorsInferMeta
  kernel:
    func: broadcast_tensors
    data_type : input
  backward: broadcast_tensors_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : build_src_rank_and_local_expert_id
  args : (Tensor expert_num_global_tensor, int64_t[] expert_num_global, int64_t num_local_experts)
  output : Tensor(vector), Tensor(local_expert_id)
  infer_meta :
    func : BuildSrcRankAndLocalExpertIdInferMeta
  kernel :
    func : build_src_rank_and_local_expert_id
    data_type : expert_num_global_tensor

- op : c_allreduce_sum
  args : (Tensor x, int ring_id, bool use_calc_stream, bool use_model_parallel)
  output : Tensor(out)
  infer_meta :
    func : AllReduceInferMeta
    param : [x]
  kernel :
    func : c_allreduce_sum
  inplace : (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : c_concat
  args : (Tensor x, int rank, int nranks, int ring_id, bool use_calc_stream, bool use_model_parallel)
  output : Tensor(out)
  infer_meta :
    func : CConcatInferMeta
    param : [x, nranks]
  kernel :
    func : c_concat
  backward: c_concat_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : c_identity
  args : (Tensor x, int ring_id, bool use_calc_stream, bool use_model_parallel)
  output : Tensor(out)
  infer_meta :
    func : CIdentityInferMeta
  kernel :
    func : c_identity
  inplace : (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : c_scatter
  args : (Tensor x, int ring_id = 0, int root = 0, int nranks = 0, bool use_calc_stream = false)
  output : Tensor(out)
  infer_meta :
    func : CScatterInferMeta
    param : [x, ring_id, root, nranks]
  kernel :
    func : c_scatter
  traits : paddle::dialect::ForwardOnlyTrait

- op : c_softmax_with_cross_entropy
  args: (Tensor logits, Tensor label,  int64_t ignore_index=-100, int ring_id=0, int rank=0, int nranks=0)
  output: Tensor(softmax), Tensor(loss)
  infer_meta:
    func : CSoftmaxWithCrossEntropyInferMeta
    spmd_rule : CSoftmaxWithCrossEntropyInferSpmd
    param: [logits, label, ignore_index, rank, nranks]
  kernel:
    func: c_softmax_with_cross_entropy
    data_type : logits
    param: [logits, label, ignore_index, rank, nranks]
  backward: c_softmax_with_cross_entropy_grad

- op : c_split
  args : (Tensor x, int rank = 0, int nranks = 1, int ring_id = 0, bool use_model_parallel = true)
  output : Tensor(out)
  infer_meta :
    func : CSplitInferMeta
    param : [x, nranks]
  kernel :
    func : c_split
    param: [x, rank, nranks, use_model_parallel]

- op : cal_aux_loss
  args : (Tensor gate_prob, Tensor dispatch_mask, Tensor tokens_mask, Tensor dispatch_tokens_mask, int64_t num_experts, bool use_group, int64_t moe_k, float clip_min)
  output : Tensor(l_aux_loss), Tensor(seqlen_float), Tensor(ce)
  infer_meta :
    func : CalAuxLossInferMeta
  kernel :
    func : cal_aux_loss
    data_type : gate_prob
  optional: tokens_mask, dispatch_tokens_mask
  backward : cal_aux_loss_grad

- op : calc_reduced_attn_scores
  args : (Tensor q, Tensor k, Tensor softmax_lse)
  output : Tensor(reduced_scores)
  infer_meta :
    func : CalcReducedAttnScoresInferMeta
    param : [q, k, softmax_lse]
  kernel :
    func : calc_reduced_attn_scores
    data_type : q
  traits : paddle::dialect::ForwardOnlyTrait

- op : cast
  args : (Tensor x, DataType dtype)
  output : Tensor(out)
  infer_meta :
    func : CastInferMeta
    spmd_rule : CastInferSpmd
  kernel :
    func : cast
    param : [x, dtype]
    data_type : x
  inplace: (x -> out)
  backward : cast_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : ceil
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : ceil
  inplace : (x -> out)
  backward : ceil_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : celu
  args : (Tensor x, float alpha = 1.0)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param: [x]
    spmd_rule : CeluInfoSpmd
  kernel :
    func : celu
  inplace : (x -> out)
  backward : celu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : channel_shuffle
  args : (Tensor x, int groups, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : ChannelShuffleInferMeta
  kernel :
    func : channel_shuffle
  backward : channel_shuffle_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : check_finite_and_unscale_
  args : (Tensor[] x, Tensor scale)
  output : Tensor[](out){x.size()}, Tensor(found_infinite)
  infer_meta :
    func : CheckFiniteAndUnscaleInferMeta
    param : [x, scale]
    spmd_rule : CheckFiniteAndUnscaleSpmd
  kernel :
    func : check_finite_and_unscale
    param : [x, scale]
    data_type : x
  inplace : (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : check_numerics
  args : (Tensor tensor, str op_type = "", str var_name = "", int check_nan_inf_level = 0, int stack_height_limit = -1, str output_dir = "")
  output : Tensor(stats), Tensor(values)
  infer_meta :
    func : CheckNumericsInferMeta
  kernel :
    func : check_numerics
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : cholesky
  args : (Tensor x, bool upper=false)
  output : Tensor
  infer_meta :
    func : CholeskyInferMeta
  kernel :
    func : cholesky
  backward : cholesky_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : cholesky_solve
  args : (Tensor x, Tensor y, bool upper=false)
  output : Tensor
  infer_meta :
    func : CholeskySolveInferMeta
  kernel :
    func : cholesky_solve
  backward : cholesky_solve_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : class_center_sample
  args : (Tensor label, int num_classes, int num_samples, int ring_id = 0, int rank = 0, int nranks = 1, bool fix_seed = false, int seed = 0)
  output : Tensor(remapped_label), Tensor(sampled_local_class_center)
  infer_meta :
    func : ClassCenterSampleInferMeta
  kernel :
    func : class_center_sample
    data_type : label
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : clip
  args : (Tensor x, Scalar(float) min, Scalar(float) max)
  output : Tensor(out)
  inplace : (x -> out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : ClipInferSpmd
  kernel :
    func : clip
    data_type : x
  backward : clip_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : clip_by_norm
  args : (Tensor x, float max_norm)
  output : Tensor(out)
  infer_meta :
    func : ClipByNormInferMeta
  kernel :
    func : clip_by_norm {dense -> dense}
           clip_by_norm_sr {selected_rows -> selected_rows}
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : coalesce_tensor
  args : (Tensor[] input, DataType dtype, bool copy_data = false, bool set_constant = false, bool persist_output = false, float constant = 0.0, bool use_align = true, int align_size = -1, int size_of_dtype = -1, int64_t[] concated_shapes = {}, int64_t[] concated_ranks = {})
  output : Tensor[](output){input.size()}, Tensor(fused_output)
  infer_meta :
    func : CoalesceTensorInferMeta
  kernel :
    func : coalesce_tensor
    data_type : dtype
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : collect_fpn_proposals
  args: (Tensor[] multi_level_rois, Tensor[] multi_level_scores, Tensor[] multi_level_rois_num,
    int post_nms_topn)
  output: Tensor (fpn_rois), Tensor (rois_num)
  infer_meta:
    func: CollectFpnProposalsInferMeta
  kernel:
    func: collect_fpn_proposals
    data_type: multi_level_rois
  optional: multi_level_rois_num, rois_num
  traits : paddle::dialect::ForwardOnlyTrait

- op : complex
  args : (Tensor real, Tensor imag)
  output : Tensor
  infer_meta :
    func : ComplexInferMeta
  kernel :
    func : complex
    data_type : real
  backward : complex_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : concat
  args : (Tensor[] x, Scalar axis=0)
  output : Tensor
  infer_meta :
    func : ConcatInferMeta
    param : [x, axis]
    spmd_rule : ConcatInferSpmdDynamic
  kernel :
    func : concat
    data_type : x
  backward : concat_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : conj
  args : (Tensor x)
  output : Tensor (out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : conj
  backward : conj_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : conv2d
  args : (Tensor input, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, str padding_algorithm="EXPLICIT", int[] dilations={1, 1}, int groups=1, str data_format="NCHW")
  output : Tensor
  infer_meta :
    func : ConvInferMeta
    spmd_rule : Conv2dInferSpmd
  kernel :
    func : conv2d
    data_type : input
  backward : conv2d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : conv2d_transpose
  args : (Tensor x, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, int[] output_padding={}, IntArray output_size={}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : Conv2dTransposeInferMeta
    spmd_rule : Conv2dTransposeInferSpmd
  kernel :
    func : conv2d_transpose
    data_type : x
  backward : conv2d_transpose_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : conv2d_transpose_bias
  args : (Tensor x, Tensor filter, Tensor bias, int[] strides={1, 1}, int[] paddings={0, 0}, int[] output_padding={}, IntArray output_size={}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : Conv2dTransposeInferMeta
    param: [x, filter, strides, paddings, output_padding, output_size, padding_algorithm, groups, dilations, data_format]
  kernel :
    func : conv2d_transpose_bias
    data_type : x
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : conv3d
  args : (Tensor input, Tensor filter, int[] strides={1, 1, 1}, int[] paddings={0, 0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1, 1}, str data_format="NCDHW")
  output : Tensor
  infer_meta :
    func : Conv3DInferMeta
    spmd_rule : Conv3dInferSpmd
  kernel :
    func : conv3d
    data_type : input
  backward : conv3d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : conv3d_transpose
  args : (Tensor x, Tensor filter, int[] strides={1, 1, 1}, int[] paddings={0, 0, 0}, int[] output_padding={}, int[] output_size={}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : ConvTransposeInferMeta
  kernel :
    func : conv3d_transpose
    data_type : x
  backward : conv3d_transpose_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : copy_to
  args : (Tensor x, Place place, bool blocking)
  output : Tensor(out)
  invoke : copy_to_impl(x, place, blocking)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : copysign
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : copysign
  inplace: (x -> out)
  backward : copysign_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : pir::BinaryElementWiseTrait

- op : correlation
  args : (Tensor input1, Tensor input2, int pad_size, int kernel_size, int max_displacement, int stride1, int stride2, int corr_type_multiply=1)
  output : Tensor(out)
  infer_meta :
    func : CorrelationInferMeta
  kernel :
    func : correlation
    data_type : input1
  backward : correlation_grad
  interfaces: paddle::dialect::LayoutTransformationInterface
  # interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : cos
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : cos
  inplace: (x -> out)
  backward : cos_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : cosh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : cosh
  inplace: (x -> out)
  backward : cosh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : crf_decoding
  args: (Tensor emission, Tensor transition, Tensor label, Tensor length)
  output: Tensor (viterbi_path)
  infer_meta:
    func: CrfDecodingInferMeta
  kernel:
    func: crf_decoding
    data_type: emission
  optional: label, length
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : crop
  args : (Tensor x, IntArray shape = {}, IntArray offsets = {})
  output : Tensor(out)
  infer_meta :
    func : CropInferMeta
  kernel :
    func : crop
    data_type : x
  backward : crop_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : cross
  args : (Tensor x, Tensor y, int axis = 9)
  output : Tensor
  infer_meta :
    func : CrossInferMeta
  kernel :
    func : cross
    data_type : x
  backward : cross_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

# Part of python API paddle.nn.functional.cross_entropy
- op : cross_entropy_with_softmax
  args : (Tensor input, Tensor label, bool soft_label=false, bool use_softmax=true, bool numeric_stable_mode=true, int ignore_index=-100, int axis=-1)
  output : Tensor(softmax), Tensor(loss)
  inplace : (input -> softmax)
  infer_meta :
    func : CrossEntropyWithSoftmaxInferMeta
    spmd_rule: CrossEntropyWithSoftmaxInferSpmd
  kernel :
    func : cross_entropy_with_softmax
    data_type : input
  backward : cross_entropy_with_softmax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : cross_entropy_with_softmax_bwd_w_downcast
  args : (Tensor label, Tensor softmax, Tensor loss_grad)
  output : Tensor(input_grad)
  infer_meta :
    func : CrossEntropyWithSoftmaxBwdWithDowncastInferMeta
  kernel :
    func : cross_entropy_with_softmax_bwd_w_downcast
    data_type : softmax

- op : ctc_align
  args: (Tensor input, Tensor input_length, int blank = 0, bool merge_repeated = true,
    int padding_value = 0)
  output: Tensor (output), Tensor (output_length)
  infer_meta:
    func: CtcAlignInferMeta
  kernel:
    func: ctc_align
    data_type: input
  optional: input_length, output_length
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : cudnn_lstm
  args: (Tensor x, Tensor init_h, Tensor init_c, Tensor w, Tensor[] weight_list, Tensor sequence_length, float dropout_prob = 0.0, bool is_bidirec = false, int hidden_size = 100, int num_layers = 1, bool is_test = false, int seed = 0)
  output: Tensor (out), Tensor (last_h), Tensor (last_c), Tensor (reserve), Tensor (state_out)
  infer_meta:
    func: CudnnLSTMInferMeta
  kernel:
    func: cudnn_lstm
    data_type: x
  optional: w, weight_list, sequence_length
  intermediate: reserve
  backward: cudnn_lstm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : cummax
  args : (Tensor x, int axis=-1, DataType dtype = DataType::INT64)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : CumWithIndicesInferMeta
    spmd_rule : CummaxInferSpmd
  kernel :
    func : cummax
    data_type : x
  backward : cummax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : cummin
  args : (Tensor x, int axis=-1, DataType dtype = DataType::INT64)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : CumWithIndicesInferMeta
    spmd_rule : CumminInferSpmd
  kernel :
    func : cummin
    data_type : x
  backward : cummin_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : cumprod
  args : (Tensor x,  int dim, bool exclusive=false, bool reverse=false)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMetaCheckAxis
    param : [x, dim]
  kernel :
    func : cumprod
  inplace: (x -> out)
  backward : cumprod_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : cumsum
  args : (Tensor x, Scalar axis=-1, bool flatten=false, bool exclusive=false, bool reverse=false)
  output : Tensor(out)
  infer_meta :
    func : CumScalarAxisInferMeta
    spmd_rule : CumSumInferSpmdDynamic
  kernel :
    func : cumsum
    data_type : x
  inplace: (x -> out)
  backward : cumsum_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : cvm
  args: (Tensor x, Tensor cvm, bool use_cvm = true)
  output: Tensor (out)
  infer_meta:
    func: CvmInferMeta
  kernel:
    func: cvm
    data_type: x
  backward: cvm_grad
  no_need_buffer: cvm

- op : data
  args : (str name, IntArray shape, DataType dtype, Place place)
  output : Tensor(out)
  infer_meta :
    func : DataInferMeta
    param : [name, shape, dtype]
  kernel:
    func : data
    param : [name, shape, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : decayed_adagrad
  args : (Tensor param, Tensor grad, Tensor moment, Tensor learning_rate, float decay = 0.95f, float epsilon = 1.0e-6f)
  output : Tensor(param_out), Tensor(moment_out)
  infer_meta :
    func : DecayedAdagradInferMeta
  kernel :
    func : decayed_adagrad
    data_type : param
  traits : paddle::dialect::ForwardOnlyTrait

- op : decode_jpeg
  args : (Tensor x, str mode, Place place)
  output : Tensor(out)
  infer_meta :
    func : DecodeJpegInferMeta
    param : [x, mode]
  kernel :
    func : decode_jpeg
    param : [x, mode]
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : deformable_conv
  args : (Tensor x, Tensor offset, Tensor filter, Tensor mask, int[] strides, int[] paddings, int[] dilations, int deformable_groups, int groups, int im2col_step)
  output : Tensor(out)
  infer_meta :
    func : DeformableConvInferMeta
  kernel :
    func : deformable_conv
    data_type : x
  optional : mask
  backward : deformable_conv_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : depend
  args: (Tensor x, Tensor[] dep)
  output: Tensor (out)
  infer_meta:
    func : UnchangedInferMeta
    param : [x]
  kernel:
    func: depend
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : depthwise_conv2d
  args : (Tensor input, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : DepthwiseConvInferMeta
    spmd_rule : DepthwiseConv2dInferSpmd
  kernel :
    func : depthwise_conv2d
    data_type : input
  backward : depthwise_conv2d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : depthwise_conv2d_bias
  args : (Tensor input, Tensor filter, Tensor bias, int[] strides={1, 1}, int[] paddings={0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : DepthwiseConv2dBiasInferMeta
  kernel :
    func : depthwise_conv2d_bias
    data_type : input
  optional : bias
  backward : depthwise_conv2d_bias_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : depthwise_conv2d_transpose
  args : (Tensor x, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, int[] output_padding={}, IntArray output_size={}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
  output : Tensor(out)
  infer_meta :
    func : Conv2dTransposeInferMeta
  kernel :
    func : depthwise_conv2d_transpose
    data_type : x
  backward : depthwise_conv2d_transpose_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : depthwise_conv3d_bias
  args : (Tensor input, Tensor filter, Tensor bias, int[] strides={1, 1, 1}, int[] paddings={0, 0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1, 1}, str data_format="NCDHW")
  output : Tensor(out)
  infer_meta :
    func : DepthwiseConv3dBiasInferMeta
  kernel :
    func : depthwise_conv3d_bias
    data_type : input
  optional : bias
  backward : depthwise_conv3d_bias_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dequantize_abs_max
  args : (Tensor x, Tensor scale, float max_range)
  output : Tensor(out)
  infer_meta :
    func : DequantizeAbsMaxInferMeta
  kernel :
    func : dequantize_abs_max
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : dequantize_log
  args: (Tensor x, Tensor dict)
  output: Tensor(out)
  infer_meta:
    func: DequantizeLogInferMeta
  kernel:
    func: dequantize_log
    data_type: x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : det
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : DetInferMeta
  kernel :
    func : determinant
  backward : det_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dgc
  args : (Tensor u, Tensor v, Tensor grad, Tensor param, Tensor current_step, Tensor nranks, float m=0.9, bool use_nesterov=true, float[] sparsity={}, float rampup_begin_step=0.0, float rampup_step=0.0, float regular_coeff=0.0, int regular_type=0)
  output : Tensor(u_out), Tensor(v_out), Tensor(encode_grad), Tensor(grad_out), Tensor(k), Tensor(gather_buff)
  infer_meta:
    func: DgcInferMeta
    param : [u, v, grad, param, current_step, nranks]
  kernel :
    func : dgc
    param : [u, v, grad, param, current_step, nranks, m, use_nesterov, sparsity, rampup_begin_step, rampup_step, regular_coeff, regular_type]
  optional: param
  data_transform :
    skip_transform : current_step, nranks
  traits : paddle::dialect::ForwardOnlyTrait

- op : dgc_clip_by_norm
  args: (Tensor x, Tensor current_step, float max_norm, float rampup_begin_step = -1.0)
  output: Tensor(out)
  infer_meta:
    func: ClipByNormInferMeta
    param: [x, max_norm]
  kernel:
    func: dgc_clip_by_norm {dense, dense -> dense}
          dgc_clip_by_norm_sr {selected_rows, dense -> selected_rows}
  data_transform :
    skip_transform : current_step
  traits : paddle::dialect::ForwardOnlyTrait

- op : dgc_momentum
  args: (Tensor param, Tensor grad, Tensor velocity, Tensor learning_rate, Tensor
    master_param, Tensor current_step_tensor, Tensor nranks_tensor, float mu, bool use_nesterov
    = false, str regularization_method = "", float regularization_coeff = 0.0f, bool
    multi_precision = false, float rescale_grad = 1.0f, float rampup_begin_step =
    -1.0)
  output: Tensor (param_out), Tensor (velocity_out), Tensor (master_param_out), Tensor
    (grad_out)
  infer_meta:
    func: DGCMomentumInferMeta
  kernel:
    func: dgc_momentum
    data_type: param
  optional : master_param, master_param_out
  data_transform :
    skip_transform : current_step_tensor, nranks_tensor
  traits : paddle::dialect::ForwardOnlyTrait

- op : diag
  args : (Tensor x, int offset = 0, float padding_value = 0.0)
  output : Tensor(out)
  infer_meta :
    func : DiagInferMeta
  kernel :
    func : diag
  backward : diag_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : diag_embed
  args : (Tensor input, int offset = 0, int dim1 = -2, int dim2 = -1)
  output : Tensor(out)
  infer_meta :
    func : DiagEmbedInferMeta
  kernel :
    func : diag_embed
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : diagonal
  args : (Tensor x, int offset = 0, int axis1 = 0, int axis2 = 1)
  output : Tensor
  infer_meta :
    func : DiagonalInferMeta
  kernel :
    func : diagonal
  backward : diagonal_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : digamma
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : digamma
  inplace: (x -> out)
  backward : digamma_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dirichlet
  args: (Tensor alpha)
  output: Tensor(out)
  infer_meta:
    func: DirichletInferMeta
  kernel:
    func: dirichlet
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : disable_check_model_nan_inf
  args: (Tensor x, int flag = 0)
  output: Tensor(out)
  infer_meta:
    func: UnchangedInferMeta
    param : [x]
  kernel:
    func: check_model_nan_inf
    data_type: x
  backward: disable_check_model_nan_inf_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dist
  args : (Tensor x, Tensor y, float p = 2.0)
  output : Tensor
  infer_meta :
    func : DistInferMeta
  kernel :
    func : dist
  backward : dist_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dot
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : DotInferMeta
  kernel :
    func : dot
    data_type : x
  backward : dot_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : dpsgd
  args: (Tensor param, Tensor grad, Tensor learning_rate, float clip = 10.0f, float batch_size = 16.0f, float sigma = 1.0f, int seed = 0)
  output: Tensor(param_out)
  infer_meta:
     func: DpsgdInferMeta
  kernel:
     func: dpsgd
     data_type: param
  traits : paddle::dialect::ForwardOnlyTrait

- op : dropout
  args : (Tensor x, Tensor seed_tensor, Scalar p = 0.5f, bool is_test = false, str mode = "downgrade_in_infer", int seed = 0, bool fix_seed = false)
  output : Tensor(out), Tensor(mask)
  infer_meta :
    func : DropoutInferMeta
    spmd_rule: DropoutFwdInferSpmd
  kernel :
    func : dropout
    data_type : x
  inplace : (x -> out)
  optional : seed_tensor
  intermediate : mask
  backward : dropout_grad
  traits : pir::SideEffectTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : edit_distance
  args : (Tensor hyps, Tensor refs, Tensor hypslength, Tensor refslength, bool normalized = false)
  output : Tensor(sequencenum), Tensor(out)
  infer_meta :
    func : EditDistanceInferMeta
  kernel :
    func : edit_distance
    data_type : DataType::FLOAT32
  optional : hypslength, refslength
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : eig
  args: (Tensor x)
  output: Tensor(out_w), Tensor(out_v)
  infer_meta:
    func: EigInferMeta
  kernel:
    func: eig
  backward: eig_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : eigh
  args : (Tensor x, str UPLO = "L")
  output : Tensor(out_w), Tensor(out_v)
  infer_meta :
    func : EighInferMeta
  kernel :
    func : eigh
  backward : eigh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : eigvals
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : EigvalsInferMeta
  kernel :
    func : eigvals
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : eigvalsh
  args : (Tensor x, str uplo = "L", bool is_test = false)
  output : Tensor(eigenvalues), Tensor(eigenvectors)
  infer_meta :
    func : EigvalshInferMeta
  kernel :
    func : eigvalsh
    data_type : x
  backward : eigvalsh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : elu
  args : (Tensor x, float alpha = 1.0f)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : EluInfoSpmd
  kernel :
    func : elu
  inplace : (x -> out)
  backward : elu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : embedding_grad_add_to
  args : (Tensor token_indices, Tensor main_grad_, Tensor out_grad)
  output : Tensor(main_grad_out)
  infer_meta :
    func : UnchangedInferMeta
    param : [main_grad_]
  kernel :
    func : embedding_grad_add_to
    param : [token_indices, main_grad_, out_grad]
    data_type : main_grad_
  inplace : (main_grad_ -> main_grad_out)

- op : embedding_with_scaled_gradient
  args : (Tensor x, Tensor weight, int64_t padding_idx=-1)
  output : Tensor
  infer_meta :
    func : EmbeddingInferMeta
    param : [x, weight, padding_idx]
  kernel :
    func : embedding {dense, dense -> dense}
    param : [x, weight, padding_idx]
    data_type : weight
  backward : embedding_with_scaled_gradient_grad

- op : empty
  args : (IntArray shape, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output: Tensor(out)
  infer_meta :
    func : CreateInferMeta
    param : [shape, dtype]
  kernel :
    func : empty
    param : [shape, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : empty_like
  args : (Tensor x, DataType dtype = DataType::UNDEFINED, Place place = {})
  output: Tensor(out)
  infer_meta :
    func : CreateLikeInferMeta
    param : [x, dtype]
    spmd_rule : EmptyLikeInferSpmd
  kernel :
    func : empty_like
    param : [x, dtype]
    data_type : dtype > x
    backend : place > x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : enable_check_model_nan_inf
  args: (Tensor x, int flag = 1)
  output: Tensor(out)
  infer_meta:
    func: UnchangedInferMeta
    param : [x]
  kernel:
    func: check_model_nan_inf
    data_type: x
  backward: enable_check_model_nan_inf_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : equal_all
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : CompareAllInferMeta
  kernel :
    func : equal_all
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : erf
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : erf
  inplace : (x -> out)
  backward : erf_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : erfinv
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : erfinv
  inplace : (x -> out)
  backward : erfinv_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : exp
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : exp
  inplace : (x -> out)
  backward : exp_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : expand
  args : (Tensor x, IntArray shape = {})
  output : Tensor(out)
  infer_meta :
    func : ExpandInferMeta
    local_shape: out
    spmd_rule : ExpandInferSpmd
  kernel :
    func : expand
    data_type : x
  backward : expand_grad

- op : expand_as
  args : (Tensor x, Tensor y, int64_t[] target_shape = {})
  output : Tensor(out)
  infer_meta :
    func : ExpandAsInferMeta
    local_shape: out
    spmd_rule : ExpandAsInferSpmd
  kernel :
    func : expand_as
    data_type : x
  optional : y
  backward : expand_as_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : expand_modality_expert_id
  args : (Tensor expert_id, int64_t num_expert_per_modality, int64_t group_size, int64_t modality_offset, bool is_group_expert)
  output : Tensor(expert_id_out)
  infer_meta :
    func : ExpandModalityExpertIdInferMeta
  kernel :
    func : expand_modality_expert_id
    data_type : expert_id

- op : expm1
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
    param : [x]
  kernel :
    func : expm1
  inplace: (x -> out)
  backward : expm1_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : exponential_
  args : (Tensor x, float lam)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : exponential
  inplace : (x -> out)
  backward : exponential__grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : eye
  args : (Scalar num_rows, Scalar num_columns, DataType dtype=DataType::FLOAT32, Place place={})
  output : Tensor(out)
  infer_meta :
    func : EyeInferMeta
    param : [num_rows, num_columns, dtype]
  kernel :
    func : eye
    param : [num_rows, num_columns, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_channel_wise_dequantize_max_abs
  args : (Tensor x, Tensor[] scales, int[] quant_bits = {8}, int quant_axis = 0, int x_num_col_dims = 1)
  output : Tensor(out)
  infer_meta :
    func : FakeChannelWiseDequantizeMaxAbsInferMeta
  kernel :
    func : fake_channel_wise_dequantize_max_abs
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_channel_wise_quantize_abs_max
  args : (Tensor x, int bit_length = 8, int round_type = 1, int quant_axis = 0, bool is_test = false)
  output : Tensor(out), Tensor(out_scale)
  infer_meta :
    func : FakeChannelWiseQuantizeAbsMaxInferMeta
  kernel :
    func : fake_channel_wise_quantize_abs_max
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_channel_wise_quantize_dequantize_abs_max
  args : (Tensor x, int bit_length = 8, int round_type = 1, int quant_axis = 0)
  output : Tensor(out), Tensor(out_scale)
  infer_meta :
    func : FakeChannelWiseQuantizeDequantizeAbsMaxInferMeta
  kernel :
    func : fake_channel_wise_quantize_dequantize_abs_max
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  backward : fake_channel_wise_quantize_dequantize_abs_max_grad

- op : fake_dequantize_max_abs
  args : (Tensor x, Tensor scale, float max_range)
  output : Tensor(out)
  infer_meta :
    func : FakeDequantizeMaxAbsInferMeta
  kernel :
    func : fake_dequantize_max_abs
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_quantize_abs_max
  args : (Tensor x, int bit_length = 8, int round_type = 1)
  output : Tensor(out), Tensor(out_scale)
  infer_meta :
    func : FakeQuantizeAbsMaxInferMeta
  kernel :
    func : fake_quantize_abs_max
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_quantize_dequantize_abs_max
  args : (Tensor x, int bit_length = 8, int round_type = 1)
  output : Tensor(out), Tensor(out_scale)
  infer_meta :
    func : FakeQuantizeAbsMaxInferMeta
  kernel :
    func : fake_quantize_dequantize_abs_max
    data_type : x
  backward : fake_quantize_dequantize_abs_max_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fake_quantize_dequantize_moving_average_abs_max
  args : (Tensor x, Tensor in_scale, Tensor in_accum, Tensor in_state, float moving_rate = 0.9, int bit_length = 8, bool is_test = false, int round_type = 1)
  output : Tensor(out), Tensor(out_scale), Tensor(out_state), Tensor(out_accum)
  infer_meta :
    func : FakeQuantOrWithDequantMovingAverageAbsMaxInferMeta
  kernel :
    func : fake_quantize_dequantize_moving_average_abs_max
    data_type : x
  optional : in_accum, in_state, out_state, out_accum
  backward : fake_quantize_dequantize_moving_average_abs_max_grad
  inplace: (in_scale -> out_scale)
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fake_quantize_moving_average_abs_max
  args : (Tensor x, Tensor in_scale, Tensor in_accum, Tensor in_state, float moving_rate = 0.9, int bit_length = 8, bool is_test = false, int round_type = 1)
  output : Tensor(out), Tensor(out_scale), Tensor(out_state), Tensor(out_accum)
  infer_meta :
    func : FakeQuantOrWithDequantMovingAverageAbsMaxInferMeta
  kernel :
    func : fake_quantize_moving_average_abs_max
    data_type : x
  optional : in_accum, in_state, out_state, out_accum
  inplace: (in_scale -> out_scale)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fake_quantize_range_abs_max
  args : (Tensor x, Tensor in_scale, Tensor iter, int window_size = 10000,  int bit_length = 8, bool is_test = false, int round_type = 1)
  output : Tensor(out), Tensor(out_scale), Tensor(out_scales)
  infer_meta :
    func : FakeQuantizeRangeAbsMaxInferMeta
  kernel :
    func : fake_quantize_range_abs_max
    data_type : x
  optional : iter, out_scales
  inplace: (in_scale -> out_scale)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fft_c2c
  args : (Tensor x, int64_t[] axes, str normalization, bool forward)
  output : Tensor
  infer_meta :
    func : FFTC2CInferMeta
  kernel :
    func : fft_c2c
  backward : fft_c2c_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fft_c2r
  args : (Tensor x, int64_t[] axes, str normalization, bool forward, int64_t last_dim_size=0L)
  output : Tensor
  infer_meta :
    func : FFTC2RInferMeta
  kernel :
    func : fft_c2r
  backward : fft_c2r_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fft_r2c
  args : (Tensor x, int64_t[] axes, str normalization, bool forward, bool onesided)
  output : Tensor
  infer_meta :
    func : FFTR2CInferMeta
  kernel :
    func : fft_r2c
  backward : fft_r2c_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fill
  args : (Tensor x, Scalar(double) value=0)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : fill
  inplace : (x -> out)
  backward: fill_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fill_diagonal
  args : (Tensor x, float value=0, int offset=0, bool wrap=false)
  output : Tensor(out)
  infer_meta :
    func : FillDiagonalInferMeta
  kernel :
    func : fill_diagonal
    data_type : x
  inplace : (x -> out)
  backward : fill_diagonal_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fill_diagonal_tensor
  args : (Tensor x, Tensor y, int64_t offset = 0, int dim1 = 0, int dim2 = 1)
  output : Tensor(out)
  infer_meta :
    func : FillDiagonalTensorInferMeta
  kernel :
    func : fill_diagonal_tensor
  inplace : (x -> out)
  backward : fill_diagonal_tensor_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flash_attn
  args : (Tensor q, Tensor k, Tensor v, Tensor fixed_seed_offset, Tensor attn_mask, float dropout = 0.0, bool causal = false, bool return_softmax = false, bool is_test = false, str rng_name = "")
  output : Tensor(out), Tensor(softmax), Tensor(softmax_lse), Tensor(seed_offset)
  optional : fixed_seed_offset, attn_mask
  infer_meta :
    func : FlashAttnInferMeta
    param : [q, k, v]
    spmd_rule : FlashAttInferSpmd
  kernel :
    func : flash_attn
    data_type : q
  backward : flash_attn_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flash_attn_qkvpacked
  args : (Tensor qkv, Tensor fixed_seed_offset, Tensor attn_mask, float dropout = 0.0, bool causal = false, bool return_softmax = false, bool is_test = false, str rng_name = "")
  output : Tensor(out), Tensor(softmax), Tensor(softmax_lse), Tensor(seed_offset)
  optional : fixed_seed_offset, attn_mask
  infer_meta :
    func : FlashAttnQKVPackedInferMeta
    param : [qkv]
  kernel :
    func : flash_attn_qkvpacked
    data_type : qkv
  backward : flash_attn_qkvpacked_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flash_attn_unpadded
  args : (Tensor q, Tensor k, Tensor v, Tensor cu_seqlens_q,  Tensor cu_seqlens_k, Tensor fixed_seed_offset, Tensor attn_mask, Scalar max_seqlen_q, Scalar max_seqlen_k, float scale, float dropout = 0.0, bool causal = false, bool return_softmax = false, bool is_test = false, str rng_name = "")
  output : Tensor(out), Tensor(softmax), Tensor(softmax_lse), Tensor(seed_offset)
  optional :  fixed_seed_offset , attn_mask
  infer_meta :
    func : FlashAttnInferMeta
    param : [q, k, v]
  kernel :
    func : flash_attn_unpadded
    data_type : q
  intermediate : softmax_lse, seed_offset
  backward : flash_attn_unpadded_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flash_attn_v3
  args : (Tensor q, Tensor k, Tensor v, Tensor q_v_, Tensor q_descale_, Tensor k_descale_, Tensor v_descale_, float softmax_scale, bool is_causal, int window_size_left, int window_size_right, float softcap, int num_splits, bool manual_set_pack_gqa, bool pack_gqa_, int sm_margin)
  output : Tensor(out), Tensor(softmax_lse)
  optional : q_v_, q_descale_, k_descale_, v_descale_
  infer_meta :
    func : FlashAttnV3InferMeta
    param : [q, k, v]
  kernel :
    func : flash_attn_v3
    data_type : q
  backward : flash_attn_v3_grad

- op : flash_attn_v3_varlen
  args : (Tensor q, Tensor k, Tensor v, Tensor cu_seqlens_q, Tensor cu_seqlens_k, Tensor seqused_q, Tensor seqused_k, Tensor qv, Tensor q_descale, Tensor k_descale, Tensor v_descale, Scalar max_seqlen_q, Scalar max_seqlen_k, float softmax_scale, bool causal, int window_size_left, int window_size_right, float softcap, int num_splits, bool manual_set_pack_gqa, bool pack_gqa, int sm_margin)
  output : Tensor(out), Tensor(softmax_lse)
  optional : seqused_q, seqused_k, qv, q_descale, k_descale, v_descale
  infer_meta :
    func : FlashAttnV3VarlenInferMeta
    param : [q, k, v]
  kernel :
    func : flash_attn_v3_varlen
    data_type : q
  backward : flash_attn_v3_varlen_grad

- op : flash_attn_varlen_qkvpacked
  args : (Tensor qkv, Tensor cu_seqlens_q,  Tensor cu_seqlens_k, Tensor fixed_seed_offset, Tensor attn_mask, Scalar max_seqlen_q, Scalar max_seqlen_k, float scale, float dropout = 0.0, bool causal = false, bool return_softmax = false, bool is_test = false, str rng_name = "", bool varlen_padded = true)
  output : Tensor(out), Tensor(softmax), Tensor(softmax_lse), Tensor(seed_offset)
  optional :  fixed_seed_offset , attn_mask
  infer_meta :
    func : FlashAttnQKVPackedInferMeta
    param : [qkv]
  kernel :
    func : flash_attn_varlen_qkvpacked
    data_type : qkv
  intermediate : softmax_lse, seed_offset
  backward : flash_attn_varlen_qkvpacked_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flashmask_attention
  args : (Tensor q, Tensor k, Tensor v, Tensor startend_row_indices,  Tensor fixed_seed_offset, float dropout = 0.0, bool causal = false, bool return_softmax = false, bool is_test = false, str rng_name = "")
  output : Tensor(out), Tensor(softmax), Tensor(softmax_lse), Tensor(seed_offset)
  optional : fixed_seed_offset
  infer_meta :
    func : FlashAttnInferMeta
    param : [q, k, v]
    spmd_rule : FlashMaskInferSpmd
  kernel :
    func : flashmask_attention
    data_type : q
  backward : flashmask_attention_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : flashmask_attention_v2
  args : (Tensor q, Tensor k, Tensor v, Tensor startend_row_indices, Tensor block_mask, float softmax_scale, bool is_causal)
  output : Tensor(out), Tensor(softmax_lse)
  optional : block_mask
  infer_meta :
    func : FlashMaskV2InferMeta
    param : [q, k, v]
  kernel :
    func : flashmask_attention_v2
    data_type : q
  backward : flashmask_attention_v2_grad

- op : flatten
  args : (Tensor x, int start_axis = 1, int stop_axis = 1)
  output : Tensor(out)
  infer_meta :
    func : FlattenInferMeta
    spmd_rule : FlattenInferSpmd
  kernel :
    func : flatten
    data_type : x
  inplace : (x -> out)
  view : (x -> out)
  backward : flatten_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : flip
  args : (Tensor x, int[] axis)
  output : Tensor (out)
  infer_meta :
    func : FlipInferMeta
  kernel :
    func : flip
  backward : flip_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : floor
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : floor
  inplace : (x -> out)
  backward : floor_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : fmax
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    param: [x, y]
    func : ElementwiseInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : fmax
  backward : fmax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : pir::BinaryElementWiseTrait

- op : fmin
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    param: [x, y]
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : fmin
  backward : fmin_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : pir::BinaryElementWiseTrait

- op : fold
  args: (Tensor x, int[] output_sizes, int[] kernel_sizes,  int[] strides, int[] paddings, int[] dilations)
  output: Tensor(out)
  infer_meta:
    func: FoldInferMeta
  kernel:
    func: fold
  backward: fold_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fractional_max_pool2d
  args : (Tensor x, int[] output_size, int[] kernel_size = {0, 0}, float random_u = 0.0, bool return_mask = true)
  output : Tensor(out), Tensor(mask)
  infer_meta :
    func : FractionalMaxPoolInferMeta
  kernel :
    func : fractional_max_pool2d
  backward : fractional_max_pool2d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : fractional_max_pool3d
  args : (Tensor x, int[] output_size, int[] kernel_size = {0, 0, 0}, float random_u = 0.0, bool return_mask = true)
  output : Tensor(out), Tensor(mask)
  infer_meta :
    func : FractionalMaxPoolInferMeta
  kernel :
    func : fractional_max_pool3d
  backward : fractional_max_pool3d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : frame
  args : (Tensor x, int frame_length, int hop_length, int axis=-1)
  output : Tensor(out)
  infer_meta :
    func : FrameInferMeta
  kernel :
    func : frame
  backward : frame_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : frobenius_norm
  args : (Tensor x, IntArray axis,  bool keep_dim,  bool reduce_all)
  output : Tensor(out)
  infer_meta :
    func : ReduceIntArrayAxisInferMetaBase
  kernel :
    func : frobenius_norm
  backward : frobenius_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : ftrl
  args: (Tensor param, Tensor squared_accumulator, Tensor linear_accumulator, Tensor grad, Tensor learning_rate, float l1=0.0f, float l2=0.0f, float lr_power=-0.5f)
  output: Tensor(param_out), Tensor(squared_accum_out), Tensor(linear_accum_out)
  infer_meta:
    func: FtrlInferMeta
  kernel:
    func: ftrl {dense, dense, dense, dense, dense -> dense, dense, dense}
          ftrl_sr {dense, dense, dense, selected_rows, dense -> dense, dense, dense}
    data_type: param
  traits : paddle::dialect::ForwardOnlyTrait

- op : full
  args : (IntArray shape, Scalar(double) value, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output: Tensor(out)
  infer_meta :
    func : CreateInferMeta
    param : [shape, dtype]
  kernel :
    func : full
    param : [shape, value, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : full_
  args : (Tensor output, IntArray shape, Scalar(double) value, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output : Tensor(out)
  inplace : (output -> out)
  infer_meta :
    func : CreateInferMeta
    param : [shape, dtype]
  kernel :
    func : full
    param : [shape, value, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : full_batch_size_like
  args : (Tensor input, int[] shape, DataType dtype, Scalar(double) value, int input_dim_idx, int output_dim_idx, Place place=CPUPlace())
  output: Tensor(out)
  infer_meta :
    func : FullBatchSizeLikeInferMeta
    param : [input, shape, value, dtype, input_dim_idx, output_dim_idx]
  kernel :
    func : full_batch_size_like
    param : [input, shape, value, dtype, input_dim_idx, output_dim_idx]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : full_int_array
  args : (int64_t[] value, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output: Tensor(out)
  infer_meta :
    func : CreateVecShapeInferMeta
    param : [value, dtype]
  kernel :
    func : full_int_array
    param : [value, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : full_like
  args : (Tensor x, Scalar value, DataType dtype = DataType::UNDEFINED, Place place = {})
  output: Tensor(out)
  infer_meta :
    func : CreateLikeInferMeta
    param : [x, dtype]
    spmd_rule : FullLikeInferSpmd
  kernel :
    func : full_like
    param : [x, value, dtype]
    data_type : dtype > x
    backend : place > x
  data_transform :
    skip_transform : x
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : full_with_tensor
  args : (Tensor value, IntArray shape, DataType dtype=DataType::FLOAT32)
  output: Tensor(out)
  infer_meta :
    func : FullWithTensorInferMeta
    param : [shape, dtype]
  kernel :
    func : full_with_tensor
    data_type : dtype
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : fused_batch_norm_act
  args : (Tensor x, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, str act_type)
  output : Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space)
  infer_meta:
    func : FusedBatchNormActInferMeta
    param : [x, scale, bias, mean, variance]
  kernel :
    func : fused_batch_norm_act
    data_type : x
  view : (mean -> mean_out), (variance -> variance_out)
  backward : fused_batch_norm_act_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fused_bn_add_activation
  args : (Tensor x, Tensor z, Tensor scale, Tensor bias, Tensor mean, Tensor variance, float momentum, float epsilon, str act_type)
  output : Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space)
  infer_meta:
    func : FusedBatchNormActInferMeta
    param : [x, scale, bias, mean, variance]
  kernel :
    func : fused_bn_add_activation
    data_type : x
  view : (mean -> mean_out), (variance -> variance_out)
  backward : fused_bn_add_activation_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fused_rms_norm_quant
  args : (Tensor x, Tensor bias, Tensor residual, Tensor norm_weight, Tensor norm_bias, float epsilon, int begin_norm_axis, float quant_scale, int quant_round_type, float quant_max_bound, float quant_min_bound)
  output : Tensor(out), Tensor(residual_out), Tensor(inv_var)
  infer_meta :
    func : FusedRmsNormQuantInferMeta
  kernel :
    func : fused_rms_norm_quant
    data_type : x
  optional : bias, residual, norm_bias, residual_out
  intermediate : inv_var
  backward : fused_rms_norm_quant_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fused_softmax_mask
  args : (Tensor x, Tensor mask)
  output : Tensor(out)
  infer_meta :
    func : SoftmaxMaskFuseInferMeta
  kernel :
    func : fused_softmax_mask
    data_type : x
  backward: fused_softmax_mask_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : fused_softmax_mask_upper_triangle
  args : (Tensor X)
  output : Tensor(Out)
  infer_meta :
    func : UnchangedInferMeta
  kernel:
    func : fused_softmax_mask_upper_triangle
  backward: fused_softmax_mask_upper_triangle_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : gammaincc
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    param : [x, y]
  kernel :
    func : gammaincc
  inplace: (x -> out)
  backward : gammaincc_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : pir::BinaryElementWiseTrait

- op : gammaln
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : gammaln
  inplace: (x -> out)
  backward : gammaln_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : gather
  args : (Tensor x, Tensor index, Scalar axis=0)
  output : Tensor(out)
  infer_meta :
    func : GatherInferMeta
  kernel :
    func : gather
    data_type: x
  backward : gather_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : gather_nd
  args : (Tensor x, Tensor index)
  output : Tensor(out)
  infer_meta :
    func : GatherNdInferMeta
    spmd_rule : GatherNdInferSpmd
  kernel :
    func : gather_nd
    data_type : x
  backward : gather_nd_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : gather_tree
  args : (Tensor ids, Tensor parents)
  output : Tensor(out)
  infer_meta :
    func : GatherTreeMeta
  kernel :
    func : gather_tree
    data_type : ids
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : gaussian
  args : (IntArray shape, float mean, float std, int seed, DataType dtype, Place place={})
  output: Tensor(out)
  infer_meta :
    func : GaussianInferMeta
    param : [shape, mean, std, seed, dtype]
  kernel :
    func : gaussian
    param : [shape, mean, std, seed, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : gaussian_inplace
  args: (Tensor x, float mean=0, float std=1.0, int seed=0)
  output: Tensor(out)
  infer_meta:
    func: UnchangedInferMeta
    param: [x]
  kernel:
    func: gaussian_inplace
    data_type: x
    backend : x
  inplace: (x -> out)
  backward: gaussian_inplace_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : gelu
  args : (Tensor x,  bool approximate = false)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param: [x]
    spmd_rule : GeluInferSpmd
  kernel :
    func : gelu
  backward : gelu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : generate_proposals
  args : (Tensor scores, Tensor bbox_deltas, Tensor im_shape, Tensor anchors, Tensor variances, int pre_nms_top_n, int post_nms_top_n, float nms_thresh, float min_size, float eta, bool pixel_offset=true)
  output : Tensor(rpn_rois), Tensor(rpn_roi_probs), Tensor(rpn_rois_num)
  infer_meta :
    func : GenerateProposalsV2InferMeta
  kernel :
    func : generate_proposals
    data_type : anchors
  optional : rpn_rois_num
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : global_gather
  args : (Tensor x, Tensor local_count, Tensor global_count, int ring_id = 0)
  output : Tensor(out)
  infer_meta:
    func : GlobalGatherInferMeta
    param: [x, local_count, global_count]
  kernel :
    func : global_gather
    data_type: x
    param: [x, local_count, global_count]
  backward : global_gather_grad

- op : global_scatter
  args : (Tensor x, Tensor local_count, Tensor global_count, int ring_id = 0)
  output : Tensor(out)
  infer_meta :
    func : GlobalScatterInferMeta
    param: [x, local_count, global_count]
  kernel :
    func : global_scatter
    data_type : x
    param: [x, local_count, global_count]
  backward : global_scatter_grad

- op : graph_khop_sampler
  args : (Tensor row, Tensor colptr, Tensor x, Tensor eids, int[] sample_sizes, bool return_eids)
  output : Tensor(out_src), Tensor(out_dst), Tensor(sample_index), Tensor(reindex_x), Tensor(out_eids)
  infer_meta :
    func : GraphKhopSamplerInferMeta
  kernel :
    func : graph_khop_sampler
    data_type : row
  optional : eids
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : graph_sample_neighbors
  args : (Tensor row, Tensor colptr, Tensor x, Tensor eids, Tensor perm_buffer, int sample_size, bool return_eids, bool flag_perm_buffer)
  output : Tensor(out), Tensor(out_count), Tensor(out_eids)
  infer_meta :
    func : GraphSampleNeighborsInferMeta
  kernel :
    func : graph_sample_neighbors
    data_type : row
  optional : eids, perm_buffer
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : grid_sample
  args : (Tensor x, Tensor grid, str mode = "bilinear", str padding_mode = "zeros", bool align_corners = true)
  output : Tensor(out)
  infer_meta :
    func : GridSampleBaseInferMeta
    param : [x, grid]
  kernel:
    func : grid_sample
    data_type : x
  backward : grid_sample_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : group_norm
  args : (Tensor x, Tensor scale, Tensor bias, float epsilon = 1e-5, int groups = -1, str data_format = "NCHW")
  output : Tensor(y), Tensor(mean), Tensor(variance)
  infer_meta :
    func : GroupNormInferMeta
    spmd_rule : GroupNormInferSpmd
  kernel :
    func : group_norm
  optional : scale, bias
  intermediate : mean, variance
  backward : group_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : gru
  args: (Tensor input, Tensor h0, Tensor weight, Tensor bias, str activation = "tanh",
    str gate_activation = "sigmoid", bool is_reverse = false, bool origin_mode = false, bool is_test=false)
  output: Tensor (batch_gate), Tensor (batch_reset_hidden_prev), Tensor (batch_hidden),
    Tensor (hidden)
  infer_meta:
    func: GruInferMeta
  kernel:
    func: gru
    data_type: input
  optional: h0, bias
  intermediate: batch_gate, batch_reset_hidden_prev, batch_hidden
  backward: gru_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : gru_unit
  args: (Tensor input, Tensor hidden_prev, Tensor weight, Tensor bias, int activation
    = 2, int gate_activation = 1, bool origin_mode = false)
  output: Tensor (gate), Tensor (reset_hidden_prev), Tensor (hidden)
  infer_meta:
    func: GruUnitInferMeta
  kernel:
    func: gru_unit
  optional: bias
  intermediate: gate, reset_hidden_prev
  backward: gru_unit_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : gumbel_softmax
  args : (Tensor x, float temperature = 1.0, bool hard = false, int axis = -1)
  output : Tensor
  infer_meta :
    func : GumbelSoftmaxInferMeta
  kernel :
    func : gumbel_softmax
  backward : gumbel_softmax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : hardshrink
  args : (Tensor x, float threshold = 0.5)
  output : Tensor (out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : hard_shrink
  backward : hardshrink_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : hardsigmoid
  args : (Tensor x, float slope = 0.2, float offset = 0.5)
  output : Tensor (out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : hardsigmoid
  inplace: (x -> out)
  backward : hardsigmoid_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : hardtanh
  args : (Tensor x, float t_min=0, float t_max=24)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : hardtanh
  inplace: (x -> out)
  backward : hardtanh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : heaviside
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
  kernel :
    func : heaviside
  backward : heaviside_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : pir::BinaryElementWiseTrait

- op : hinge_loss
  args: (Tensor logits, Tensor labels)
  output: Tensor (loss)
  infer_meta:
    func: HingeLossInferMeta
  kernel:
    func: hinge_loss
    data_type: logits
  backward: hinge_loss_grad

- op : histogram
  args : (Tensor input, Tensor weight, int64_t bins = 100, float min = 0.0, float max = 0.0, bool density = false)
  output : Tensor(out)
  infer_meta :
    func : HistogramInferMeta
  optional : weight
  kernel :
    func : histogram
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : hsigmoid_loss
  args : (Tensor x, Tensor label, Tensor w, Tensor bias, Tensor path, Tensor code, int num_classes, bool is_sparse)
  output : Tensor(out), Tensor(pre_out), Tensor(w_out)
  infer_meta :
    func : HSigmoidLossInferMeta
  optional: path, code, bias
  kernel :
    func : hsigmoid_loss
    data_type : x
  backward : hsigmoid_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : huber_loss
  args : (Tensor input, Tensor label, float delta)
  output : Tensor(out), Tensor(residual)
  infer_meta :
    func : HuberLossInferMeta
  kernel :
    func : huber_loss
  intermediate : residual
  backward : huber_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : i0
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : i0
  inplace: (x -> out)
  backward : i0_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : i0e
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : i0e
  backward : i0e_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : i1
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : i1
  backward : i1_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : i1e
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : i1e
  backward : i1e_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : identity_loss
  args : (Tensor x, int reduction = 1)
  output : Tensor(out)
  infer_meta :
    func : IdentityLossInferMeta
  kernel :
    func : identity_loss
  inplace: (x -> out)
  backward : identity_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : im2sequence
  args: (Tensor x, Tensor y, int[] kernels, int[] strides = {1, 1}, int[] paddings
    = {0, 0, 0, 0}, int[] out_stride = {1, 1})
  output: Tensor (out)
  infer_meta:
    func: Im2sequenceInferMeta
  kernel:
    func: im2sequence
  optional: y
  backward: im2sequence_grad

- op : imag
  args : (Tensor x)
  output : Tensor (out)
  infer_meta :
    func : RealAndImagInferMeta
  kernel :
    func : imag
  backward : imag_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : increment
  args : (Tensor x, float value = 1.0)
  output : Tensor(out)
  infer_meta :
    func : IncrementInferMeta
  kernel :
    func : increment
  inplace : (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : index_add
  args : (Tensor x, Tensor index,  Tensor add_value, int axis = 0)
  output : Tensor(out)
  infer_meta :
    func : IndexAddInferMeta
  kernel :
    func : index_add
    data_type : x
  inplace : (x -> out)
  backward : index_add_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : index_elementwise_get
  args : (Tensor x, Tensor[] index, int64_t[] input_dims, int64_t[] input_strides, int64_t[] index_dims, int64_t[] index_stride, int64_t slice_offset = 0, bool accumulate = true, bool is_combined = false)
  output : Tensor (out)
  infer_meta :
    func : IndexElementwiseGetInferMeta
  kernel :
    func : index_elementwise_get
    data_type : x
  backward : index_elementwise_get_grad

- op : index_elementwise_put
  args : (Tensor x, Tensor[] index, Scalar value, int64_t[] input_dims, int64_t[] input_strides, int64_t[] index_dims, int64_t[] index_strides, int64_t slice_offset)
  output : Tensor (out)
  infer_meta :
    func : IndexElementwisePutInferMeta
  kernel :
    func : index_elementwise_put
    data_type : x
  inplace : (x -> out)
  backward : index_elementwise_put_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : index_elementwise_put_with_tensor
  args : (Tensor x, Tensor[] index, Tensor value, int64_t[] input_dims, int64_t[] input_strides, int64_t[] index_dims, int64_t[] index_strides, int64_t slice_offset)
  output : Tensor (out)
  infer_meta :
    func : IndexElementwisePutWithTensorInferMeta
  kernel :
    func : index_elementwise_put_with_tensor
    data_type : x
  inplace : (x -> out)
  backward : index_elementwise_put_with_tensor_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : index_put
  args : (Tensor x, Tensor[] indices, Tensor value, bool accumulate=false)
  output : Tensor(out)
  infer_meta :
    func : IndexPutInferMeta
    spmd_rule : IndexPutInferSpmd
  kernel :
    func : index_put
    data_type : x
  inplace : (x -> out)
  backward : index_put_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : index_sample
  args : (Tensor x, Tensor index)
  output : Tensor
  infer_meta :
    func : IndexSampleInferMeta
  kernel :
    func : index_sample
    data_type : x
  backward : index_sample_grad
  data_transform :
    skip_transform : index
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : index_select
  args : (Tensor x, Tensor index, int axis = 0)
  output : Tensor(out)
  infer_meta :
    func : IndexSelectInferMeta
    spmd_rule : IndexSelectInferSpmd
  kernel :
    func : index_select
    data_type : x
  backward : index_select_grad
  data_transform :
    skip_transform : index
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : index_select_strided
  args : (Tensor x, int64_t index, int axis = 0)
  output : Tensor(out)
  infer_meta :
    func : IndexSelectStridedInferMeta
  kernel :
    func : index_select_strided
    data_type : x
  backward : index_select_strided_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : instance_norm
  args : (Tensor x, Tensor scale, Tensor bias, float epsilon=1e-5)
  output : Tensor(y), Tensor(saved_mean), Tensor(saved_variance)
  infer_meta :
    func : InstanceNormInferMeta
    spmd_rule : InstanceNormInferSpmd
  kernel :
    func : instance_norm
    data_type : x
  optional : scale, bias
  intermediate : saved_mean, saved_variance
  backward : instance_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : interp_antialias
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : interp_antialias
    data_type : x
  backward : interp_antialias_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : inverse
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : InverseInferMeta
  kernel :
    func : inverse
  backward : inverse_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : is_empty
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : IsEmptyInferMeta
  kernel :
    func : is_empty
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : isclose
  args : (Tensor x, Tensor y, Scalar(double) rtol=1e-5, Scalar(double) atol=1e-8,  bool equal_nan=false)
  output : Tensor(out)
  infer_meta :
    func : ValueCompareInferMeta
    param: [x, y]
  kernel :
    func : isclose
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : isfinite
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : IsfiniteInferMeta
  kernel :
    func : isfinite {dense -> dense},
           isfinite_sr {selected_rows -> selected_rows}
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : isinf
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : IsfiniteInferMeta
  kernel :
    func : isinf {dense -> dense},
           isinf_sr {selected_rows -> selected_rows}
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : isnan
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : IsfiniteInferMeta
  kernel :
    func : isnan {dense -> dense},
           isnan_sr {selected_rows -> selected_rows}
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : kldiv_loss
  args : (Tensor x, Tensor label, str reduction = "mean", bool log_target = false)
  output : Tensor(out)
  infer_meta :
    func : KLDivInferMeta
  kernel :
    func : kldiv_loss
    data_type : x
  backward : kldiv_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : kron
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : KronInferMeta
  kernel :
    func : kron
  backward : kron_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : kthvalue
  args : (Tensor x, int64_t k = 1, int axis = -1, bool keepdim = false)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : KthvalueInferMeta
  kernel :
    func : kthvalue
  backward : kthvalue_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : l1_norm
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : L1NormInferMeta
  kernel :
    func : l1_norm
    data_type : x
  inplace: (x -> out)
  backward : l1_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : label_smooth
  args : (Tensor label, Tensor prior_dist, float epsilon = 0.0f)
  output : Tensor (out)
  infer_meta :
    func : UnchangedInferMeta
    param : [label]
    spmd_rule : LabelSmoothInferSpmd
  kernel :
    func : label_smooth
    data_type : label
  optional : prior_dist
  backward : label_smooth_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lamb_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment1, Tensor moment2, Tensor beta1_pow, Tensor beta2_pow, Tensor master_param, Tensor skip_update, float weight_decay, float beta1=0.9, float beta2=0.999, float epsilon=1.0e-6f, bool always_adapt=false, bool multi_precision=false)
  output : Tensor(param_out), Tensor(moment1_out), Tensor(moment2_out), Tensor(beta1_pow_out), Tensor(beta2_pow_out), Tensor(master_param_outs)
  infer_meta :
    func : LambInferMeta
  kernel :
    func : lamb {dense, dense, dense, dense, dense, dense, dense, dense, dense -> dense, dense, dense, dense, dense, dense},
           lamb_sr {dense, selected_rows, dense, dense, dense, dense, dense, dense, dense -> dense, dense, dense, dense, dense, dense}
    data_type : param
  optional : master_param, skip_update, beta1_pow_out, beta2_pow_out, master_param_outs
  inplace : (param -> param_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (beta1_pow -> beta1_pow_out), (beta2_pow -> beta2_pow_out), (master_param -> master_param_outs)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : layer_norm
  args : (Tensor x, Tensor scale, Tensor bias, float epsilon = 1e-5, int begin_norm_axis = 1)
  output : Tensor(out), Tensor(mean), Tensor(variance)
  infer_meta :
    func : LayerNormInferMeta
    spmd_rule : LayerNormInferSpmd
  kernel :
    func : layer_norm
    data_type : x
  backward : layer_norm_grad
  intermediate : mean, variance
  optional : scale, bias
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : leaky_relu
  args : (Tensor x, double negative_slope = 0.02)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : leaky_relu
  inplace: (x -> out)
  backward : leaky_relu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : lerp
  args : (Tensor x, Tensor y, Tensor weight)
  output : Tensor(out)
  infer_meta :
    func : LerpInferMeta
  kernel :
    func : lerp
    data_type : x
  inplace : (x -> out)
  backward : lerp_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lgamma
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule: ElementwiseUnaryInferSpmd
  kernel :
    func : lgamma
  inplace: (x -> out)
  backward : lgamma_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits: pir::UnaryElementWiseTrait

- op : limit_by_capacity
  args : (Tensor expert_count, Tensor capacity, int n_worker)
  output : Tensor(out)
  infer_meta :
    func : LimitByCapacityInferMeta
  kernel :
    func : limit_by_capacity
    data_type : expert_count
  traits : paddle::dialect::ForwardOnlyTrait

- op : linear_interp
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : linear_interp
    data_type : x
  backward : linear_interp_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : linear_v2
  args : (Tensor input, Tensor weight, Tensor bias, bool transpose_weight=false)
  output : Tensor(out)
  infer_meta :
    func : LinearV2InferMeta
    spmd_rule : LinearV2InferSpmd
  kernel :
    func : linear_v2
    data_type : input
  backward : linear_v2_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : linspace
  args : (Tensor start, Tensor stop, Tensor number, DataType dtype, Place place)
  output : Tensor(out)
  infer_meta :
    func : LinspaceInferMeta
    param: [start, stop, number, dtype]
  kernel :
    func : linspace
    param: [start, stop, number, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : llm_int8_linear
  args : (Tensor x, Tensor weight, Tensor bias, Tensor weight_scale, float threshold=6.0)
  output : Tensor(out)
  infer_meta :
    func : LLMInt8LinearInferMeta
  kernel :
    func : llm_int8_linear
    data_type : x
  optional: bias
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : log
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule: ElementwiseUnaryInferSpmd
  kernel :
    func : log
  inplace: (x -> out)
  backward: log_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : log10
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : log10
  inplace: (x -> out)
  backward: log10_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : log1p
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : log1p
  inplace: (x -> out)
  backward: log1p_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : log2
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : log2
  inplace: (x -> out)
  backward: log2_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : log_loss
  args : (Tensor input, Tensor label, float epsilon)
  output : Tensor
  infer_meta :
    func : LogLossInferMeta
  kernel :
    func : log_loss
  backward : log_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : log_softmax
  args : (Tensor x, int axis = -1)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMetaCheckAxis
    spmd_rule : SoftmaxInferSpmd
  kernel :
    func : log_softmax
    data_type : x
  backward : log_softmax_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : logcumsumexp
  args : (Tensor x, int axis=-1, bool flatten=false, bool exclusive=false, bool reverse=false)
  output : Tensor(out)
  infer_meta :
    func : CumInferMeta
  kernel :
    func : logcumsumexp
  backward : logcumsumexp_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : logical_and
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : LogicalBinaryInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : logical_and
    data_type : x
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : logical_not
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : LogicalNotInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : logical_not
    data_type : x
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::UnaryElementWiseTrait

- op : logical_or
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : LogicalBinaryInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : logical_or
    data_type : x
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : logical_xor
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : LogicalBinaryInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : logical_xor
    data_type : x
    backend : x
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : logit
  args : (Tensor x, double eps = 1e-6)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : LogitInfoSpmd
  kernel :
    func : logit
  inplace: (x -> out)
  backward : logit_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : logsigmoid
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : logsigmoid
  backward : logsigmoid_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : logspace
  args : (Tensor start, Tensor stop, Tensor num, Tensor base, DataType dtype, Place place={})
  output : Tensor(out)
  infer_meta:
    func : LogspaceInferMeta
    param : [start, stop, num, base, dtype]
  kernel :
    func : logspace
    param : [start, stop, num, base, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : logsumexp
  args : (Tensor x, int[] axis={},  bool keepdim=false,  bool reduce_all=false)
  output : Tensor(out)
  infer_meta :
    func : LogsumexpInferMeta
    spmd_rule : LogSumExpInferSpmd
  kernel :
    func : logsumexp
  backward : logsumexp_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : lookup_table_dequant
  args: (Tensor w, Tensor ids, int64_t padding_idx = -1)
  output: Tensor (out)
  infer_meta:
    func: LookupTableDequantInferMeta
  kernel:
    func: lookup_table_dequant
    data_type: w
  traits : paddle::dialect::ForwardOnlyTrait

- op : lp_pool2d
  args : (Tensor x, IntArray kernel_size, int64_t[] strides = {1,1}, int64_t[] paddings = {0,0}, bool ceil_mode = false, bool exclusive = true, str data_format = "NCHW", str pooling_type = "", bool global_pooling = false, bool adaptive = false, str padding_algorithm = "EXPLICIT", float norm_type = 0.0f)
  output : Tensor(out)
  infer_meta :
    func : Pool2DInferMeta
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm]
  kernel :
    func : lp_pool2d
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm, norm_type]
  backward : lp_pool2d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lstm
  args: (Tensor input, Tensor h0, Tensor c0, Tensor weight, Tensor bias, bool use_peepholes
    = true, bool is_reverse = false, bool is_test = false, str gate_activation = "sigmoid",
    str cell_activation = "tanh", str candidate_activation = "tanh")
  output: Tensor (hidden), Tensor (cell), Tensor (batch_gate), Tensor (batch_cell_pre_act)
  infer_meta:
    func: LSTMInferMeta
  kernel:
    func: lstm
    data_type: input
  optional: h0, c0
  intermediate: batch_gate, batch_cell_pre_act
  backward: lstm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lstsq
  args : (Tensor x, Tensor y, Scalar rcond=0.0f, str driver="gels")
  output : Tensor(solution), Tensor(residuals), Tensor(rank), Tensor(singular_values)
  infer_meta :
    func : LstsqInferMeta
  kernel :
    func : lstsq
    data_type : x
  optional : residuals
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lu
  args : (Tensor x, bool pivot = true)
  output : Tensor(out), Tensor(pivots), Tensor(infos)
  infer_meta :
    func : LUInferMeta
  kernel :
    func : lu
    data_type : x
  inplace : (x -> out)
  backward : lu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lu_solve
  args : (Tensor b, Tensor lu, Tensor pivots, str trans)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [b]
  kernel :
    func : lu_solve
    data_type : b
  backward : lu_solve_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : lu_unpack
  args : (Tensor x, Tensor y, bool unpack_ludata = true, bool unpack_pivots = true)
  output : Tensor(pmat), Tensor(l), Tensor(u)
  infer_meta :
    func : LUUnpackInferMeta
  kernel :
    func : lu_unpack
    data_type : x
  backward : lu_unpack_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : margin_cross_entropy
  args : (Tensor logits, Tensor label, bool return_softmax = false, int ring_id = 0, int rank = 0, int nranks = 1, float margin1 = 1.0f, float margin2 = 0.5f, float margin3 = 0.0f, float scale = 64.0f)
  output : Tensor(softmax), Tensor(loss)
  infer_meta :
    func : MarginCrossEntropyInferMeta
  kernel :
    func : margin_cross_entropy
    data_type : logits
  backward : margin_cross_entropy_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : masked_fill
  args : (Tensor x, Tensor mask, Tensor value)
  output : Tensor (out)
  inplace: (x -> out)
  infer_meta :
    func : MaskedFillInferMeta
  kernel :
    func : masked_fill
    data_type : x
  backward : masked_fill_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : masked_multihead_attention_
  args : (Tensor x, Tensor cache_kv, Tensor bias, Tensor src_mask, Tensor cum_offsets, Tensor sequence_lengths, Tensor rotary_tensor, Tensor beam_cache_offset, Tensor qkv_out_scale, Tensor out_shift, Tensor out_smooth, int seq_len, int rotary_emb_dims, bool use_neox_rotary_style=false, str compute_dtype = "default", float out_scale=-1, int quant_round_type=1, float quant_max_bound=127.0, float quant_min_bound=-127.0)
  output : Tensor(out), Tensor(cache_kv_out), Tensor(beam_cache_offset_out)
  infer_meta :
    func : MaskedMultiheadAttentionInferMeta
  kernel :
    func : masked_multihead_attention
    data_type : x
  optional : bias, src_mask, cum_offsets, sequence_lengths, rotary_tensor, beam_cache_offset, qkv_out_scale, out_shift, out_smooth
  inplace : (cache_kv -> cache_kv_out), (beam_cache_offset -> beam_cache_offset_out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : masked_select
  args : (Tensor x, Tensor mask)
  output : Tensor (out)
  infer_meta :
    func : MaskedSelectInferMeta
  kernel :
    func : masked_select
    data_type : x
  backward : masked_select_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : match_matrix_tensor
  args: (Tensor x, Tensor y, Tensor w, int dim_t = 1)
  output: Tensor (out), Tensor (tmp)
  infer_meta:
    func: MatchMatrixTensorInferMeta
  kernel:
    func: match_matrix_tensor
  backward: match_matrix_tensor_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : matrix_nms
  args : (Tensor bboxes, Tensor scores, float score_threshold, int nms_top_k, int keep_top_k, float post_threshold=0., bool use_gaussian = false, float gaussian_sigma = 2., int background_label = 0, bool normalized = true)
  output : Tensor(out), Tensor(index), Tensor(roisnum)
  infer_meta :
    func : MatrixNMSInferMeta
  optional : roisnum
  kernel :
    func : matrix_nms
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : matrix_power
  args : (Tensor x, int n)
  output : Tensor
  infer_meta :
    func : MatrixPowerInferMeta
  kernel :
    func : matrix_power
  backward : matrix_power_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : matrix_rank
  args : (Tensor x, float tol, bool use_default_tol=true, bool hermitian=false)
  output : Tensor(out)
  infer_meta :
    func : MatrixRankInferMeta
    param : [x, use_default_tol, hermitian]
  kernel :
    func : matrix_rank
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : matrix_rank_atol_rtol
  args : (Tensor x, Tensor atol, Tensor rtol, bool hermitian=false)
  output : Tensor(out)
  infer_meta :
    func : MatrixRankAtolRtolInferMeta
  kernel :
    func : matrix_rank_atol_rtol
  optional : rtol
  traits : paddle::dialect::ForwardOnlyTrait

- op : matrix_rank_tol
  args : (Tensor x, Tensor atol_tensor, bool use_default_tol=true, bool hermitian=false)
  output : Tensor(out)
  infer_meta :
    func : MatrixRankTolInferMeta
  kernel :
    func : matrix_rank_tol
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : max
  args : (Tensor x, IntArray axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : StrictReduceIntArrayAxisInferMeta
    spmd_rule: ReductionMaxInferSpmdDynamic
  kernel :
    func : max
  backward : max_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : max_pool2d_with_index
  args : (Tensor x, int[] kernel_size, int[] strides= {1, 1}, int[] paddings = {0, 0}, int[] dilations = {1, 1}, bool global_pooling = false, bool adaptive = false, bool ceil_mode = false)
  output : Tensor(out), Tensor(mask)
  infer_meta :
    func : MaxPoolWithIndexInferMeta
  kernel :
    func : max_pool2d_with_index
  backward : max_pool2d_with_index_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : max_pool3d_with_index
  args : (Tensor x, int[] kernel_size, int[] strides = {1, 1, 1}, int[] paddings = {0, 0, 0}, int[] dilations = {1, 1, 1}, bool global_pooling = false, bool adaptive = false, bool ceil_mode = false)
  output : Tensor(out), Tensor(mask)
  infer_meta :
    func : MaxPoolWithIndexInferMeta
  kernel :
    func : max_pool3d_with_index
  backward : max_pool3d_with_index_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : max_with_index
  args : (Tensor x, Scalar(int64_t) dim, bool keepdim = false, bool flatten = false)
  output : Tensor(values), Tensor(indices)
  infer_meta :
    func : MinMaxWithIndexInferMeta
  kernel :
    func : max_with_index
    data_type : x
  backward : max_with_index_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : maxout
  args : (Tensor x, int groups, int axis = 1)
  output : Tensor(out)
  infer_meta :
    func : MaxOutInferMeta
  kernel :
    func : maxout
  backward : maxout_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : mean
  args : (Tensor x, IntArray axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceIntArrayAxisInferMeta
    spmd_rule : ReductionMeanInferSpmdDynamic
  kernel :
    func : mean
  backward : mean_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : mean_all
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : MeanAllInferMeta
    spmd_rule : MeanAllInferSpmd
  kernel :
    func : mean_all
  backward : mean_all_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : median
  args : (Tensor x, IntArray axis = {}, bool keepdim = true, str mode="avg")
  output : Tensor(out), Tensor(medians)
  infer_meta :
    func : MedianInferMeta
  kernel :
    func : median
  backward : median_grad

- op : memcpy_d2h
  args : (Tensor x, int dst_place_type)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : memcpy_d2h
  traits : paddle::dialect::ForwardOnlyTrait

- op : memcpy_h2d
  args : (Tensor x, int dst_place_type)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : memcpy_h2d
  traits : paddle::dialect::ForwardOnlyTrait

- op : memory_efficient_attention
  args : (Tensor query, Tensor key, Tensor value, Tensor bias, Tensor cu_seqlens_q, Tensor cu_seqlens_k, Tensor causal_diagonal, Tensor seqlen_k, Scalar max_seqlen_q, Scalar max_seqlen_k, bool causal, double dropout_p, float scale, bool is_test)
  output : Tensor(output), Tensor(logsumexp), Tensor(seed_and_offset)
  infer_meta :
    func : MemoryEfficientAttentionInferMeta
  kernel :
    func : memory_efficient_attention
    data_type : query
  optional : bias, cu_seqlens_q, cu_seqlens_k, causal_diagonal, seqlen_k
  backward : memory_efficient_attention_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : merge_selected_rows
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : merge_selected_rows {selected_rows -> selected_rows}
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : merged_adam_
  args : (Tensor[] param, Tensor[] grad, Tensor[] learning_rate, Tensor[] moment1, Tensor[] moment2, Tensor[] moment2_max, Tensor[] beta1_pow, Tensor[] beta2_pow, Tensor[] master_param, Scalar beta1 = 0.9f, Scalar beta2 = 0.999f, Scalar epsilon = 1.0e-8f, bool multi_precision = false, bool use_global_beta_pow = false, bool amsgrad = false)
  output : Tensor[](param_out){param.size()}, Tensor[](moment1_out){param.size()}, Tensor[](moment2_out){param.size()}, Tensor[](moment2_max_out){param.size()}, Tensor[](beta1_pow_out){param.size()}, Tensor[](beta2_pow_out){param.size()}, Tensor[](master_param_out){param.size()}
  infer_meta :
    func : MergedAdamInferMeta
  kernel :
    func : merged_adam
    data_type : param
  optional: moment2_max, master_param, moment2_max_out, master_param_out
  inplace : (param -> param_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (moment2_max -> moment2_max_out), (beta1_pow -> beta1_pow_out), (beta2_pow -> beta2_pow_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : merged_momentum_
  args : (Tensor[] param, Tensor[] grad, Tensor[] velocity, Tensor[] learning_rate, Tensor[] master_param, float mu, bool use_nesterov = false, str[] regularization_method = {}, float[] regularization_coeff = {}, bool multi_precision = false, float rescale_grad = 1.0f)
  output : Tensor[](param_out){param.size()}, Tensor[](velocity_out){param.size()}, Tensor[](master_param_out){param.size()}
  infer_meta :
    func : MergedMomentumInferMeta
  kernel :
    func : merged_momentum
    data_type : param
  optional: master_param, master_param_out
  inplace : (param -> param_out), (velocity -> velocity_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : meshgrid
  args : (Tensor[] inputs)
  output : Tensor[](out){inputs.size()}
  infer_meta :
    func : MeshgridInferMeta
  kernel :
    func : meshgrid
    data_type : inputs
  backward : meshgrid_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : min_with_index
  args : (Tensor x, Scalar(int64_t) dim, bool keepdim = false, bool flatten = false)
  output : Tensor(values), Tensor(indices)
  infer_meta :
    func : MinMaxWithIndexInferMeta
  kernel :
    func : min_with_index
    data_type : x
  backward : min_with_index_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : mish
  args : (Tensor x, float lambda)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : MishInfoSpmd
  kernel :
    func : mish
  inplace: (x -> out)
  backward : mish_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : mode
  args : (Tensor x,  int axis = -1,  bool keepdim = false)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : ModeInferMeta
  kernel :
    func : mode
  backward : mode_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : moe_combine
  args : (Tensor x, Tensor combine_weights, Tensor scatter_index)
  output : Tensor(y)
  infer_meta :
    func : MoeCombineInferMeta
  kernel :
    func : moe_combine
    data_type : x
  backward : moe_combine_grad

- op : moe_combine_auto
  args : (Tensor x, Tensor combine_weights, Tensor scatter_index)
  output : Tensor(y)
  infer_meta :
    func : MoeCombineInferMeta
    spmd_rule : MoECombineInferSpmd
  kernel :
    func : moe_combine
    data_type : x
  backward : moe_combine_auto_grad

- op : moe_combine_no_weight
  args : (Tensor x, Tensor combine_weight, Tensor scatter_index, float epsilon = 1.0e-15)
  output : Tensor(y)
  infer_meta :
    func : MoeCombineNoWeightInferMeta
  kernel :
    func : moe_combine_no_weight
    data_type : x
  backward : moe_combine_no_weight_grad

- op : moe_gate_dispatch
  args : (Tensor x, Tensor gate_logits, Tensor corr_bias, int64_t k, int64_t capacity, bool use_pad)
  output : Tensor(y), Tensor(combine_weights), Tensor(scatter_index), Tensor(expert_offset), Tensor(expert_id)
  infer_meta :
    func : MoeGateDispatchInferMeta
  kernel :
    func : moe_gate_dispatch
    data_type : x
  optional : corr_bias
  backward : moe_gate_dispatch_grad

- op : moe_gate_dispatch_and_quant
  args : (Tensor x, Tensor gate_logits, Tensor corr_bias, int64_t k, int64_t capacity, bool use_pad, bool use_pow2_scale)
  output : Tensor(out_fp8), Tensor(scale), Tensor(combine_weights), Tensor(scatter_index), Tensor(expert_offset), Tensor(expert_id)
  infer_meta :
    func : MoeGateDispatchAndQuantInferMeta
  kernel :
    func : moe_gate_dispatch_and_quant
    data_type : x
  optional : corr_bias

- op : moe_gate_dispatch_auto
  args : (Tensor x, Tensor gate_logits, Tensor corr_bias, int64_t k, int64_t capacity, bool use_pad)
  output : Tensor(y), Tensor(combine_weights), Tensor(scatter_index), Tensor(expert_offset), Tensor(expert_id)
  infer_meta :
    func : MoeGateDispatchAutoInferMeta
    spmd_rule : MoEGateDispatchInferSpmd
  kernel :
    func : moe_gate_dispatch
    data_type : x
  optional : corr_bias
  backward : moe_gate_dispatch_auto_grad

- op : moe_gate_dispatch_partial_nosoftmaxtopk
  args : (Tensor x, Tensor combine_weights, Tensor expert_id, int64_t k, int64_t capacity, int64_t num_experts, bool use_pad, int64_t expert_start_index, int64_t expert_end_index, bool reverse_token_drop)
  output : Tensor(y), Tensor(combine_weights_out), Tensor(scatter_index), Tensor(scatter_index_rev), Tensor(expert_offset), Tensor(expert_nums_local)
  infer_meta :
    func : MoeGateDispatchPartialNoSoftmaxTopKInferMeta
  kernel :
    func : moe_gate_dispatch_partial_nosoftmaxtopk
    data_type : x
  # inplace : (combine_weights -> combine_weights_out)
  backward : moe_gate_dispatch_partial_nosoftmaxtopk_grad

- op : moe_gate_dispatch_permute
  args : (Tensor x, Tensor gate_logits, Tensor corr_bias, int64_t k, int64_t capacity, int64_t world_size)
  output : Tensor(y), Tensor(combine_weights), Tensor(scatter_index), Tensor(expert_offset), Tensor(expert_id)
  infer_meta :
    func : MoeGateDispatchPermuteInferMeta
  kernel :
    func : moe_gate_dispatch_permute
    data_type : x
  optional : corr_bias
  backward : moe_gate_dispatch_permute_grad

- op : moe_permute
  args : (Tensor hidden_states, Tensor scale, Tensor expert_routemap_topk, Tensor expert_prob_topk, int num_experts, int[] tokens_per_expert, int padding_alignment, bool do_gather, bool using_ue8m0_scale = false, bool return_expert_indices=false, int override_buffer_size = -1)
  output : Tensor(hidden_states_unzipped), Tensor(zipped_expertwise_rowmap), Tensor(token_prob_unzipped), Tensor(scale_unzipped), Tensor(expert_indices)
  infer_meta:
    func : MoePermuteInferMeta
  kernel :
    func : moe_permute
    data_type : hidden_states
  optional : scale

- op : moe_unpermute
  args : (Tensor hidden_states_unzipped, Tensor zipped_expertwise_rowmap, Tensor expert_routemap_topk, Tensor token_prob_unzipped, int total_zipped_tokens_num, int num_experts, bool use_mix_precision, bool using_weighted_combine=false)
  output : Tensor(hidden_states), Tensor(expert_prob_topk)
  infer_meta :
    func : MoeUnpermuteInferMeta
  kernel :
    func : moe_unpermute
    data_type : hidden_states_unzipped

- op : momentum_
  args : (Tensor param, Tensor grad, Tensor velocity, Tensor learning_rate, Tensor master_param, float mu, bool use_nesterov = false, str regularization_method = "", float regularization_coeff = 0.0f, bool multi_precision = false, float rescale_grad = 1.0f)
  output : Tensor(param_out), Tensor(velocity_out), Tensor(master_param_out)
  infer_meta:
    func : MomentumInferMeta
  kernel :
    func : momentum {dense, dense, dense, dense, dense -> dense, dense, dense},
           momentum_dense_param_sparse_grad {dense, selected_rows, dense, dense, dense -> dense, dense, dense}
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (velocity -> velocity_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : mp_allreduce_sum
  args : (Tensor x, int ring_id = 0)
  output : Tensor(out)
  infer_meta :
    func : AllReduceInferMeta
    param: [x]
  kernel :
    func : mp_allreduce_sum
    param: [x]
  backward: mp_allreduce_sum_grad
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : multi_dot
  args : (Tensor[] x)
  output : Tensor
  infer_meta :
    func : MultiDotInferMeta
  kernel :
    func : multi_dot
  backward : multi_dot_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : multiclass_nms3
  args : (Tensor bboxes, Tensor scores, Tensor rois_num, float score_threshold, int nms_top_k, int keep_top_k, float nms_threshold=0.3, bool normalized=true, float nms_eta=1.0, int background_label=0)
  output : Tensor(out), Tensor(index), Tensor(nms_rois_num)
  infer_meta :
    func : MultiClassNMSInferMeta
  kernel :
    func : multiclass_nms3
    data_type : scores
  optional : rois_num, nms_rois_num
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : multinomial
  args : (Tensor x, Scalar(int) num_samples = 1, bool replacement = false)
  output : Tensor(out)
  infer_meta :
    func : MultinomialInferMeta
  kernel :
    func : multinomial
    data_type : x
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : multiplex
  args : (Tensor[] inputs, Tensor index)
  output : Tensor
  infer_meta :
    func : MultiplexInferMeta
  kernel :
    func : multiplex
    data_type : inputs
  backward : multiplex_grad
  data_transform :
    skip_transform : index
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : mv
  args : (Tensor x, Tensor vec)
  output : Tensor
  infer_meta :
    func : MvInferMeta
  kernel :
    func : mv
  backward : mv_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : nadam_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor momentum_decay_pow, Tensor beta2_pow, Tensor mu_product, Tensor moment1, Tensor moment2, Tensor master_param, float beta1 = 0.9f, float beta2 = 0.999f, float epsilon = 1.0e-8f, float momentum_decay = 0.004f, bool multi_precision = false)
  output : Tensor(param_out), Tensor(momentum_decay_pow_out), Tensor(beta2_pow_out), Tensor(mu_product_out), Tensor(moment1_out), Tensor(moment2_out), Tensor(master_param_out)
  infer_meta :
    func : NAdamInferMeta
  kernel :
    func : nadam
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (momentum_decay_pow -> momentum_decay_pow_out), (beta2_pow -> beta2_pow_out), (mu_product -> mu_product_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (master_param->master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : nanmedian
  args : (Tensor x, IntArray axis = {}, bool keepdim = true, str mode="avg")
  output : Tensor(out), Tensor(medians)
  infer_meta :
    func : NanmedianInferMeta
  kernel :
    func : nanmedian
  backward : nanmedian_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : nearest_interp
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : nearest_interp
    data_type : x
  backward : nearest_interp_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : nextafter
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    param: [x, y]
  kernel :
    func : nextafter
    data_type : x
  traits : paddle::dialect::ForwardOnlyTrait, pir::BinaryElementWiseTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : nll_loss
  args : (Tensor input, Tensor label, Tensor weight, int64_t ignore_index = -100, str reduction = "mean")
  output : Tensor(out), Tensor(total_weight)
  infer_meta :
    func : NllLossRawInferMeta
  kernel :
    func : nll_loss
    data_type : input
  optional : weight
  backward : nll_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : nms
  args : (Tensor x, float threshold = 1.0f)
  output : Tensor(out)
  infer_meta :
    func : NMSInferMeta
  kernel :
    func : nms
    data_type : x
  traits : paddle::dialect::ForwardOnlyTrait

- op : nonzero
  args : (Tensor condition)
  output : Tensor(out)
  infer_meta :
    func : NonZeroInferMeta
    spmd_rule : NonZeroInferSpmd
  kernel :
    func : nonzero
    data_type: condition
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : norm
  args : (Tensor x, int axis, float epsilon, bool is_test)
  output : Tensor(out), Tensor(norm)
  infer_meta :
    func : NormInferMeta
  kernel :
    func : norm
  backward : norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : npu_identity
  args : (Tensor x, int format = -1)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : npu_identity
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : numel
  args : (Tensor x)
  output : Tensor(size)
  infer_meta :
    func : NumelInferMeta
    spmd_rule : NumelInferSpmd
  kernel :
    func : numel
    data_type : x
  data_transform:
    skip_transform : x
  no_need_buffer : x
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : one_hot
  args : (Tensor x, Scalar(int) num_classes)
  output : Tensor(out)
  infer_meta :
    func : OneHotInferMeta
    spmd_rule : OneHotInferSpmdDynamic
  kernel :
    func : one_hot
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : ones
  args : (IntArray shape, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output : Tensor(out)
  invoke : full(shape, 1, dtype, place)
  traits : paddle::dialect::ForwardOnlyTrait

- op : ones_like
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Place place={})
  output : Tensor(out)
  invoke : full_like(x, 1, dtype, place)
  traits : paddle::dialect::ForwardOnlyTrait

- op : overlap_add
  args: (Tensor x, int hop_length, int axis=-1)
  output: Tensor
  infer_meta:
    func: OverlapAddInferMeta
  kernel:
    func: overlap_add
    data_type : x
  backward: overlap_add_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : p_norm
  args : (Tensor x,  float porder=2,  int axis=-1,  float epsilon=1.0e-12f,  bool keepdim=false,  bool asvector=false)
  output : Tensor(out)
  infer_meta :
    func : PNormInferMeta
    spmd_rule: PNormInferSpmd
  kernel :
    func : p_norm
  backward : p_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : pad
  args : (Tensor x, int[] paddings, Scalar pad_value)
  output : Tensor
  infer_meta :
    func : PadInferMeta
    spmd_rule : PadInferSpmdDynamic
  kernel :
    func : pad
  backward : pad_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : pad3d
  args : (Tensor x, IntArray paddings, str mode = "constant", double pad_value = 0.0, str data_format = "NCDHW")
  output : Tensor(out)
  infer_meta :
    func : Pad3dInferMeta
  kernel :
    func : pad3d
  backward : pad3d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : partial_allgather
  args : (Tensor x, int nranks, int rank, int ring_id = 0)
  output : Tensor(out)
  infer_meta :
    func: PartialAllgatherInferMeta
    param: [x, nranks, rank]
  kernel :
    func : partial_allgather
    param: [x, nranks, rank]
  inplace : (x -> out)

- op : partial_concat
  args : (Tensor[] x, int start_index = 0, int length = -1)
  output : Tensor(out)
  infer_meta :
    func : PartialConcatInferMeta
  kernel :
    func : partial_concat
    data_type : x
  backward : partial_concat_grad

- op : partial_sum
  args : (Tensor[] x, int start_index = 0, int length = -1)
  output : Tensor(out)
  infer_meta :
    func : PartialSumInferMeta
  kernel :
    func : partial_sum
    data_type : x
  backward : partial_sum_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : pixel_shuffle
  args : (Tensor x, int upscale_factor=1, str data_format="NCHW")
  output : Tensor
  infer_meta :
    func : PixelShuffleInferMeta
  kernel :
    func : pixel_shuffle
  backward : pixel_shuffle_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : pixel_unshuffle
  args : (Tensor x, int downscale_factor=1, str data_format="NCHW")
  output : Tensor
  infer_meta :
    func : PixelUnshuffleInferMeta
  kernel :
    func : pixel_unshuffle
  backward : pixel_unshuffle_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : poisson
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule: ElementwiseUnaryInferSpmd
  kernel :
    func : poisson
  backward : poisson_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : polygamma
  args : (Tensor x, int n)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param: [x]
  kernel :
    func : polygamma
  inplace: (x -> out)
  backward : polygamma_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : pool2d
  args : (Tensor x, IntArray kernel_size, int64_t[] strides, int64_t[] paddings, bool ceil_mode, bool exclusive, str data_format, str pooling_type, bool global_pooling, bool adaptive, str padding_algorithm)
  output : Tensor(out)
  infer_meta :
    func : Pool2DInferMeta
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm]
  kernel :
    func : pool2d
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm]
  backward : pool2d_grad
  interfaces : paddle::dialect::LayoutTransformationInterface, paddle::dialect::InferSymbolicShapeInterface

- op : pool3d
  args : (Tensor x, int64_t[] kernel_size, int64_t[] strides, int64_t[] paddings, bool ceil_mode, bool exclusive, str data_format, str pooling_type, bool global_pooling, bool adaptive, str padding_algorithm)
  output : Tensor(out)
  infer_meta :
    func : PoolInferMeta
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm]
  kernel :
    func : pool3d
    param : [x, kernel_size, strides, paddings, ceil_mode, exclusive, data_format, pooling_type, global_pooling, adaptive, padding_algorithm]
  backward : pool3d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : pow
  args : (Tensor x, Scalar y=1.0f)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param: [x]
    spmd_rule: PowInferSpmd
  kernel :
    func : pow
    data_type : x
  inplace: (x -> out)
  backward : pow_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : prelu
  args : (Tensor x, Tensor alpha, str data_format="NCHW", str mode="all")
  output : Tensor(out)
  infer_meta :
    func : PReluInferMeta
  kernel :
    func : prelu
    data_type : x
  backward : prelu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : prior_box
  args : (Tensor input, Tensor image, float[] min_sizes, float[] max_sizes = {}, float[] aspect_ratios = {}, float[] variances = {}, bool flip=true, bool clip=true, float step_w=0.0, float step_h=0.0, float offset=0.5, bool min_max_aspect_ratios_order=false)
  output : Tensor(out), Tensor(var)
  infer_meta :
    func : PriorBoxInferMeta
  kernel :
    func : prior_box
    data_type : input
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : prod
  args : (Tensor x, IntArray axis, bool keepdim, bool reduce_all)
  output : Tensor
  infer_meta :
    func : ReduceIntArrayAxisInferMetaBase
  kernel :
    func : prod
  backward : prod_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : prune_gate_by_capacity
  args : (Tensor gate_idx, Tensor expert_count, int64_t n_expert=0, int64_t n_worker=0)
  output : Tensor(out_gate_idx)
  infer_meta :
    func : PruneGateByCapacityInferMeta
  kernel :
    func : prune_gate_by_capacity
    data_type : gate_idx
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : psroi_pool
  args : (Tensor x, Tensor boxes, Tensor boxes_num, int pooled_height=1, int pooled_width=1, int output_channels=1, float spatial_scale=1.0)
  output : Tensor
  infer_meta :
    func : PsroiPoolInferMeta
  kernel :
    func : psroi_pool
    data_type : x
  optional : boxes_num
  backward : psroi_pool_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : put_along_axis
  args : (Tensor arr, Tensor indices, Tensor values, int axis, str reduce = "assign", bool include_self = true)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [arr]
    spmd_rule : PutAlongAxisInferSpmd
  kernel :
    func : put_along_axis
    data_type : arr
  inplace : (arr -> out)
  backward : put_along_axis_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : pyramid_hash
  args: (Tensor x, Tensor w, Tensor white_list, Tensor black_list, int num_emb = 0,
    int space_len = 0, int pyramid_layer = 2, int rand_len = 0, float drop_out_percent
    = 0, int is_training = 0, bool use_filter = true, int white_list_len = 0, int
    black_list_len = 0, int seed = 0, float lr = 0.0, str distribute_update_vars =
    "")
  output: Tensor (out), Tensor (drop_pos), Tensor (x_temp_out)
  infer_meta:
    func: PyramidHashInferMeta
  kernel:
    func: pyramid_hash
    data_type: w
  intermediate: x_temp_out
  backward: pyramid_hash_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : qr
  args : (Tensor x, str mode = "reduced")
  output : Tensor(q), Tensor(r)
  infer_meta :
    func : QrInferMeta
  kernel :
    func : qr
  backward : qr_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : radam_
  args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor beta1_pow, Tensor beta2_pow, Tensor rho, Tensor moment1, Tensor moment2, Tensor master_param, float beta1 = 0.9f, float beta2 = 0.999f, float epsilon = 1.0e-8f, bool multi_precision = false)
  output : Tensor(param_out), Tensor(beta1_pow_out), Tensor(beta2_pow_out), Tensor(rho_out), Tensor(moment1_out), Tensor(moment2_out), Tensor(master_param_out)
  infer_meta :
    func : RAdamInferMeta
  kernel :
    func : radam
    data_type : param
  optional : master_param, master_param_out
  inplace : (param -> param_out), (beta1_pow -> beta1_pow_out), (beta2_pow -> beta2_pow_out), (rho -> rho_out), (moment1 -> moment1_out), (moment2 -> moment2_out), (master_param->master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : randint
  args : (int low, int high, IntArray shape, DataType dtype=DataType::INT64, Place place={})
  output : Tensor(out)
  infer_meta :
    func : RandintInferMeta
    param : [low, high, shape, dtype]
  kernel :
    func : randint
    param : [low, high, shape, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : random
  args : (Tensor x, int64_t from, int64_t to)
  output : Tensor(out)
  infer_meta :
    func : RandomInferMeta
    param : [x]
  kernel :
    func : random
  inplace : (x -> out)
  backward: random_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : pir::SideEffectTrait

- op : random_routing
  args : (Tensor prob, Tensor topk_value, Tensor topk_idx)
  output : Tensor(out)
  infer_meta :
    func : RandomRoutingInferMeta
  kernel :
    func : random_routing
    data_type : prob
  inplace : (topk_idx -> out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : randperm
  args : (int n, DataType dtype, Place place={})
  output : Tensor(out)
  infer_meta :
    func : RandpermInferMeta
    param : [n, dtype]
  kernel :
    func : randperm
    param : [n, dtype]
    data_type : dtype
    backend : place
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : rank_attention
  args : (Tensor x, Tensor rank_offset, Tensor rank_param, int max_rank = 3, int max_size = 0)
  output : Tensor(input_help), Tensor(out), Tensor(ins_rank)
  infer_meta :
    func : RankAttentionInferMeta
  kernel :
    func : rank_attention
    data_type : x
  backward : rank_attention_grad
  optional : ins_rank, input_help
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : read_file
  args : (str filename = "", DataType dtype=DataType::UINT8, Place place=CPUPlace())
  output : Tensor(out)
  infer_meta :
    func : ReadFileInferMeta
    param : [filename]
  kernel :
    func : read_file
    param : [filename]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : real
  args : (Tensor x)
  output : Tensor (out)
  infer_meta :
    func : RealAndImagInferMeta
  kernel :
    func : real
  backward : real_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : reciprocal
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : reciprocal
  inplace : (x -> out)
  backward : reciprocal_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : reduce
  args : (Tensor x, int ring_id = 0, int root_id = 0, int reduce_type = 0)
  output : Tensor(out)
  infer_meta :
    func : DistReduceInferMeta
    param: [x]
  kernel :
    func : reduce
    param: [x, root_id, reduce_type]
  traits : paddle::dialect::ForwardOnlyTrait
  inplace : (x -> out)

- op : reduce_as
  args : (Tensor x, Tensor target)
  output : Tensor(out)
  infer_meta :
    func : ReduceAsInferMeta
  kernel :
    func : reduce_as
    data_type : x
  backward : reduce_as_grad
  no_need_buffer : target
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : reduce_scatter
  args : (Tensor x, int ring_id = 0, int nranks = 1)
  output : Tensor(out)
  infer_meta :
    func : ReduceScatterInferMeta
    param: [x, nranks]
  kernel :
    func : reduce_scatter
    param: [x, nranks]
  traits : paddle::dialect::ForwardOnlyTrait

- op : reindex_graph
  args : (Tensor x, Tensor neighbors, Tensor count, Tensor hashtable_value, Tensor hashtable_index)
  output : Tensor(reindex_src), Tensor(reindex_dst), Tensor(out_nodes)
  infer_meta :
    func : GraphReindexInferMeta
  kernel :
    func : graph_reindex
    data_type : x
  optional : hashtable_value, hashtable_index
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : relu
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : relu
  inplace : (x -> out)
  backward : relu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : relu6
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : relu6
  backward : relu6_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : renorm
  args : (Tensor x, float p, int axis, float max_norm)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : renorm
  inplace: (x -> out)
  backward : renorm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : repeat_interleave
  args : (Tensor x, int repeats, int axis, int64_t output_size = -1)
  output : Tensor(out)
  infer_meta :
    func : RepeatInterleaveInferMeta
  kernel :
    func : repeat_interleave
    data_type : x
  backward: repeat_interleave_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : repeat_interleave_with_tensor_index
  args : (Tensor x, Tensor repeats, int axis, int64_t output_size = -1)
  output : Tensor(out)
  infer_meta :
    func : RepeatInterleaveWithTensorIndexInferMeta
  kernel :
    func : repeat_interleave_with_tensor_index
    data_type : x
  backward: repeat_interleave_with_tensor_index_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : reshape
  args : (Tensor x, IntArray shape)
  output : Tensor(out)
  infer_meta :
    func : ReshapeInferMeta
    spmd_rule : ReshapeInferSpmd
    local_shape: out
    global_shape: out
  kernel :
    func : reshape
  inplace : (x -> out)
  view: (x -> out)
  backward: reshape_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : restrict_nonzero
  args : (Tensor condition, int64_t total_true_num)
  output : Tensor (out)
  infer_meta :
    func : RestrictNonZeroInferMeta
  kernel :
    func : restrict_nonzero
  traits : paddle::dialect::ForwardOnlyTrait

- op : reverse
  args : (Tensor x, IntArray axis)
  output : Tensor
  infer_meta :
    func : ReverseInferMeta
  kernel :
    func : reverse
    data_type : x
  backward : reverse_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : rint
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : rint
  inplace : (x -> out)
  backward : rint_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : rmsprop_
  args : (Tensor param, Tensor mean_square, Tensor grad, Tensor moment, Tensor learning_rate, Tensor mean_grad, Tensor master_param, float epsilon = 1.0e-10f, float decay = 0.9f, float momentum = 0.0f, bool centered = false, bool multi_precision = false)
  output : Tensor(param_out), Tensor(moment_out), Tensor(mean_square_out), Tensor(mean_grad_out), Tensor(master_param_outs)
  infer_meta :
    func : RmspropInferMeta
  kernel :
    func : rmsprop {dense, dense, dense, dense, dense, dense, dense-> dense, dense, dense, dense, dense}
           rmsprop_dense_param_sparse_grad {dense, dense, selected_rows, dense, dense, dense, dense-> dense, dense, dense, dense, dense}
    data_type : param
  optional : mean_grad, master_param, master_param_outs
  inplace : (param -> param_out), (moment -> moment_out), (mean_square -> mean_square_out), (mean_grad -> mean_grad_out), (master_param->master_param_outs)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : rnn
  args: (Tensor x, Tensor[] pre_state, Tensor[] weight_list, Tensor sequence_length, Tensor dropout_state_in, float dropout_prob=0.0, bool is_bidirec=false, int input_size=10, int hidden_size=100, int num_layers=1, str mode="RNN_TANH", int seed=0, bool is_test=false)
  output: Tensor(out), Tensor(dropout_state_out), Tensor[](state){pre_state.size()}, Tensor(reserve)
  infer_meta:
    func: RnnInferMeta
    param : [x, pre_state, weight_list, sequence_length, dropout_prob, is_bidirec, input_size, hidden_size, num_layers, mode, seed, is_test]
  kernel:
    func: rnn
    param : [x, pre_state, weight_list, sequence_length, dropout_prob, is_bidirec, input_size, hidden_size, num_layers, mode, seed, is_test]
    data_type: x
  backward: rnn_grad
  optional : sequence_length
  intermediate : reserve
  view : (dropout_state_in -> dropout_state_out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : roi_align
  args : (Tensor x, Tensor boxes, Tensor boxes_num, int pooled_height=1, int pooled_width=1, float spatial_scale=1.0, int sampling_ratio=-1, bool aligned=false)
  output : Tensor
  infer_meta :
    func : RoiAlignInferMeta
    spmd_rule : RoiAlignInferSpmd
  kernel :
    func : roi_align
    data_type : x
  optional : boxes_num
  backward : roi_align_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : roi_pool
  args : (Tensor x, Tensor boxes, Tensor boxes_num, int pooled_height=1, int pooled_width=1, float spatial_scale=1.0)
  output : Tensor(out), Tensor(arg_max)
  infer_meta :
    func : RoiPoolInferMeta
  kernel :
    func : roi_pool
    data_type : x
  optional : boxes_num
  intermediate : arg_max
  backward : roi_pool_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : roll
  args : (Tensor x, IntArray shifts={}, int64_t[] axis={})
  output : Tensor(out)
  infer_meta :
    func : RollInferMeta
    spmd_rule : RollInferSpmdDynamic
  kernel :
    func : roll
    data_type : x
  backward : roll_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : round
  args : (Tensor x, int decimals = 0 )
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : RoundInfoSpmd
  kernel :
    func : round
  inplace : (x -> out)
  backward : round_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : rprop_
  args : (Tensor param, Tensor grad, Tensor prev, Tensor learning_rate, Tensor master_param, Tensor learning_rate_range, Tensor etas, bool multi_precision=false)
  output : Tensor(param_out), Tensor(prev_out), Tensor(learning_rate_out), Tensor(master_param_out)
  infer_meta :
    func : RpropInferMeta
  kernel :
    func : rprop
    data_type : param
  data_transform :
    support_trans_dtype : learning_rate
  optional : master_param, master_param_out
  inplace : (param -> param_out), (prev -> prev_out), (learning_rate -> learning_rate_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : rrelu
  args : (Tensor x, float lower=1.0f/8, float upper=1.0f/3, bool is_test=false)
  output : Tensor(out), Tensor(noise)
  infer_meta :
    func : RReluInferMeta
  kernel :
    func : rrelu
    data_type : x
  inplace: (x -> out)
  intermediate : noise
  backward : rrelu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : rsqrt
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : rsqrt
  inplace : (x -> out)
  backward : rsqrt_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : scale
  args : (Tensor x, Scalar scale=1.0, Scalar bias=0.0, bool bias_after_scale=true)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : ScaleInferSpmd
  kernel :
    func : scale {dense -> dense},
           scale_sr {selected_rows -> selected_rows}
    data_type : x
  inplace : (x -> out)
  backward : scale_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : scatter
  args : (Tensor x, Tensor index, Tensor updates, bool overwrite=true)
  output : Tensor(out)
  infer_meta :
    func : ScatterInferMeta
    spmd_rule : ScatterInferSpmd
  kernel :
    func : scatter
    data_type : x
  inplace : (x -> out)
  backward : scatter_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : scatter_nd_add
  args : (Tensor x, Tensor index, Tensor updates)
  output : Tensor
  infer_meta :
    func : ScatterNdAddInferMeta
    spmd_rule : ScatterNdAddInferSpmd
  kernel :
    func : scatter_nd_add
    data_type : x
  backward : scatter_nd_add_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : searchsorted
  args : (Tensor sorted_sequence, Tensor values, bool out_int32 = false, bool right = false)
  output : Tensor(out)
  infer_meta :
    func : SearchsortedInferMeta
  kernel :
    func : searchsorted
    data_type : sorted_sequence
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : segment_pool
  args : (Tensor x, Tensor segment_ids, str pooltype="SUM")
  output : Tensor(out), Tensor(summed_ids)
  infer_meta :
    func : SegmentPoolInferMeta
  kernel :
    func : segment_pool
    data_type : x
  intermediate : summed_ids
  backward : segment_pool_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : selu
  args : (Tensor x, float scale=1.0507009873554804934193349852946, float alpha=1.6732632423543772848170429916717)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : SeluInfoSpmd
  kernel :
    func : selu
  inplace: (x -> out)
  backward : selu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : send_u_recv
  args : (Tensor x, Tensor src_index, Tensor dst_index, str reduce_op = "SUM", IntArray out_size = {0})
  output : Tensor(out), Tensor(dst_count)
  infer_meta :
    func : SendURecvInferMeta
  kernel :
    func : send_u_recv
    data_type : x
  intermediate : dst_count
  backward : send_u_recv_grad

- op : send_ue_recv
  args : (Tensor x, Tensor y, Tensor src_index, Tensor dst_index, str message_op="ADD", str reduce_op="SUM", IntArray out_size={0})
  output : Tensor(out), Tensor(dst_count)
  infer_meta :
    func : SendUERecvInferMeta
  kernel :
    func : send_ue_recv
    data_type : x
  intermediate : dst_count
  backward : send_ue_recv_grad

- op : send_uv
  args : (Tensor x, Tensor y, Tensor src_index, Tensor dst_index, str message_op = "ADD")
  output : Tensor(out)
  infer_meta :
    func : SendUVInferMeta
  kernel :
    func : send_uv
    data_type : x
  backward : send_uv_grad

- op : sequence_conv
  args: (Tensor x, Tensor padding_data, Tensor filter, int context_length, bool padding_trainable = false,
    int context_start = 0, int context_stride = 1)
  output: Tensor (out)
  infer_meta:
    func: SequenceConvInferMeta
  kernel:
    func: sequence_conv
    data_type: x
  optional: padding_data
  backward: sequence_conv_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : sequence_mask
  args: (Tensor x, Scalar(int) max_len, DataType out_dtype)
  output: Tensor(y)
  infer_meta:
    func: SequenceMaskScalarInferMeta
  kernel:
    func: sequence_mask_scalar
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : sequence_pool
  args: (Tensor x, bool is_test=false, str pooltype = "AVERAGE", float pad_value = 0.0)
  output: Tensor (out), Tensor (max_index)
  infer_meta:
    func: SequencePoolInferMeta
  kernel:
    func: sequence_pool
  intermediate: max_index
  backward: sequence_pool_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : set
  args : (Tensor x, Tensor source, int64_t[] dims = {}, int64_t[] stride = {}, int64_t offset = 0)
  output : Tensor (out)
  infer_meta :
    func : SetInferMeta
    param : [x, dims, stride]
  kernel :
    func : set
  inplace : (x -> out)

- op : set_value_with_tensor
  args : (Tensor x, Tensor values, IntArray starts, IntArray ends, IntArray steps, int64_t[] axes, int64_t[] decrease_axes, int64_t[] none_axes)
  output : Tensor(out)
  inplace: (x -> out)
  infer_meta:
    func: SetValueInferMeta
    param: [x]
  kernel:
    func: set_value_with_tensor
  backward: set_value_with_tensor_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : sgd_
  args : (Tensor param, Tensor learning_rate, Tensor grad, Tensor master_param, bool multi_precision=false)
  output : Tensor(param_out), Tensor(master_param_out)
  infer_meta :
    func : SgdInferMeta
    spmd_rule : SgdInferSpmd
  kernel :
    func : sgd {dense, dense, dense, dense -> dense, dense},
           sgd_dense_param_sparse_grad {dense, dense, selected_rows, dense -> dense, dense},
           sgd_sparse_param_sparse_grad {selected_rows, dense, selected_rows, selected_rows -> selected_rows, selected_rows}
    data_type : param
  data_transform :
    support_trans_dtype : learning_rate
  optional : master_param, master_param_out
  inplace : (param -> param_out), (master_param -> master_param_out)
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : shape
  args : (Tensor input)
  output : Tensor(out)
  infer_meta :
    func : ShapeInferMeta
  kernel :
    func : shape {dense -> dense},
           shape_sr {selected_rows -> dense}
  data_transform :
    skip_transform : input
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait
  no_need_buffer : input

- op : shape64
  args : (Tensor input)
  output : Tensor(out)
  infer_meta :
    func : Shape64InferMeta
  kernel :
    func : shape64 {dense -> dense},
           shape64_sr {selected_rows -> dense}
  data_transform :
    skip_transform : input
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait
  no_need_buffer : input

- op : shard_index
  args : (Tensor input, int index_num, int nshards, int shard_id, int ignore_value=-1)
  output : Tensor(out)
  infer_meta :
    func : ShardIndexInferMeta
  kernel :
    func : shard_index
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : share_data
  args: (Tensor x)
  output: Tensor (out)
  infer_meta:
    func: ShareDataInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel:
    func: share_data {dense -> dense}
          share_data_sr {selected_rows -> selected_rows}
  traits : paddle::dialect::ForwardOnlyTrait

- op : shuffle_batch
  args : (Tensor x, Tensor seed, int startup_seed=0)
  output : Tensor(out), Tensor(shuffle_idx), Tensor(seed_out)
  infer_meta:
     func: ShuffleBatchInferMeta
  kernel:
     func: shuffle_batch
     data_type: x
  backward : shuffle_batch_grad
  traits : pir::SideEffectTrait
  data_transform :
    skip_transform : seed
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : shuffle_channel
  args : (Tensor x, int group = 1)
  output : Tensor(out)
  infer_meta :
    func : ShuffleChannelInferMeta
  kernel :
    func : shuffle_channel
  backward : shuffle_channel_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : sigmoid
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : sigmoid
  inplace : (x -> out)
  backward : sigmoid_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : sigmoid_cross_entropy_with_logits
  args : (Tensor x, Tensor label, Tensor pos_weight, bool normalize=false, int ignore_index=-100)
  output : Tensor
  infer_meta :
    func : SigmoidCrossEntropyWithLogitsInferMeta
  kernel :
    func : sigmoid_cross_entropy_with_logits
  inplace : (x -> out)
  backward : sigmoid_cross_entropy_with_logits_grad
  optional : pos_weight
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : sign
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : sign
  backward : sign_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits: pir::UnaryElementWiseTrait

- op : silu
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : silu
  inplace : (x -> out)
  backward : silu_grad
  interfaces : paddle::dialect::LayoutTransformationInterface, paddle::dialect::InferSymbolicShapeInterface
  traits: pir::UnaryElementWiseTrait

- op : sin
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : sin
  inplace : (x -> out)
  backward : sin_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : sinh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : sinh
  inplace: (x -> out)
  backward : sinh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : slice
  args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)
  output : Tensor
  infer_meta :
    func : SliceRawInferMeta
    spmd_rule : SliceInferSpmdDynamic
  kernel :
    func : slice
  backward : slice_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : slogdet
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : slogdet
  backward : slogdet_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : slogdet_v2
  args : (Tensor x)
  output : Tensor(sign), Tensor(logdet)
  infer_meta :
    func : SlogdetV2InferMeta
  kernel :
    func : slogdet_v2
  backward : slogdet_v2_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : slow_conv2d_dilated
  args : (Tensor input, Tensor filter, Tensor bias, int[] strides={1, 1}, int[] paddings={0, 0}, str padding_algorithm="EXPLICIT", int[] dilations={1, 1}, int groups=1, str data_format="NCHW")
  output : Tensor
  infer_meta :
    func : SlowConvDilatedInferMeta
  kernel :
    func : slow_conv2d_dilated
    data_type : input
  optional : bias
  backward : slow_conv2d_dilated_grad
  interfaces :  paddle::dialect::LayoutTransformationInterface

- op : slow_conv3d_dilated
  args : (Tensor input, Tensor filter, Tensor bias, int[] strides={1, 1, 1}, int[] paddings={0, 0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1, 1}, str data_format="NCDHW")
  output : Tensor
  infer_meta :
    func : SlowConv3DDilatedInferMeta
  kernel :
    func : slow_conv3d_dilated
    data_type : input
  optional : bias
  backward : slow_conv3d_dilated_grad

- op : softplus
  args : (Tensor x, double beta = 1.0, double threshold = 20.0)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : SoftplusInfoSpmd
  kernel :
    func : softplus
  backward : softplus_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : softshrink
  args : (Tensor x, float threshold = 0.5)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : SoftshrinkInfoSpmd
  kernel :
    func : softshrink
  backward : softshrink_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : softsign
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
    param : [x]
  kernel :
    func : softsign
  backward : softsign_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : solve
  args : (Tensor x, Tensor y)
  output : Tensor
  infer_meta :
    func : SolveInferMeta
  kernel :
    func : solve
    data_type : x
  backward : solve_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : sparse_attention
  args: (Tensor q, Tensor k, Tensor v, Tensor offset, Tensor columns, Tensor key_padding_mask,
    Tensor attn_mask)
  output: Tensor (out), Tensor (sparse_dot_sdd), Tensor (softmax)
  infer_meta:
    func: SparseAttentionInferMeta
  kernel:
    func: sparse_attention
    data_type: q
  optional: key_padding_mask, attn_mask
  intermediate: sparse_dot_sdd, softmax
  backward: sparse_attention_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : spectral_norm
  args : (Tensor weight, Tensor u, Tensor v, int dim = 0, int power_iters = 1, float eps = 1e-12f)
  output : Tensor
  infer_meta :
    func : SpectralNormInferMeta
  kernel :
    func : spectral_norm
    data_type : weight
  backward : spectral_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : split
  args : (Tensor x, IntArray sections, Scalar(int) axis)
  output : Tensor[]{sections.size()}
  infer_meta :
    func : SplitInferMeta
    spmd_rule : SplitInferSpmdDynamic
  kernel :
    func : split
  backward : split_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : split_with_num
  args : (Tensor x, int num, Scalar(int) axis)
  output : Tensor[]{num}
  infer_meta :
    func : SplitWithNumInferMeta
    spmd_rule : SplitWithNumInferSpmdDynamic
  kernel :
    func : split_with_num
  backward : split_with_num_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : sqrt
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : sqrt {dense -> dense},
           sqrt_sr {selected_rows -> selected_rows}
  inplace : (x -> out)
  backward : sqrt_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : square
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : square {dense -> dense},
           square_sr {selected_rows -> selected_rows}
  inplace : (x -> out)
  backward : square_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : squared_l2_norm
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : SquaredL2NormInferMeta
    spmd_rule : SquaredL2NormInferSpmd
  kernel :
    func : squared_l2_norm
  backward : squared_l2_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : squeeze
  args : (Tensor x, IntArray axis={})
  output : Tensor(out)
  infer_meta :
    func : SqueezeInferMeta
    spmd_rule : SqueezeInferSpmd
  kernel :
    func : squeeze
    data_type : x
  inplace : (x -> out)
  view: (x -> out)
  backward : squeeze_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : stack
  args : (Tensor[] x, int axis = 0)
  output : Tensor (out)
  infer_meta :
    func : StackInferMeta
    spmd_rule : StackInferSpmd
  kernel :
    func : stack
  backward : stack_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : standard_gamma
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
  kernel :
    func : standard_gamma
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : stanh
  args : (Tensor x, float scale_a=0.67f, float scale_b=1.7159f)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : StanhInfoSpmd
  kernel :
    func : stanh
  backward : stanh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : std
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false, bool unbiased=true, double correction=1)
  output : Tensor(out)
  infer_meta :
    func : StdInferMeta
  kernel :
    func : std
    data_type : x
  backward : std_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : stft
  args: (Tensor x, Tensor window, int n_fft, int hop_length, bool normalized, bool onesided)
  output: Tensor (out)
  infer_meta:
    func: StftInferMeta
  kernel:
    func: stft
    data_type: x
  backward: stft_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : strided_slice
  args : (Tensor x, int[] axes, IntArray starts, IntArray ends, IntArray strides)
  output : Tensor
  infer_meta :
    func : StridedSliceInferMeta
    spmd_rule : StridedSliceInferSpmdDynamic
  kernel :
    func : strided_slice
  backward : strided_slice_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : sum
  args : (Tensor x, IntArray axis={}, DataType dtype=DataType::UNDEFINED, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : SumInferMeta
    spmd_rule : ReductionSumInferSpmdDynamic
  kernel :
    func : sum
    data_type : x
  backward : sum_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : svd
  args : (Tensor x, bool full_matrices = false)
  output : Tensor(u), Tensor(s), Tensor(vh)
  infer_meta :
    func : SvdInferMeta
  kernel :
    func : svd
  backward : svd_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : svdvals
  args : (Tensor x)
  output :  Tensor(s)
  infer_meta :
    func : SvdvalsInferMeta
  kernel :
    func : svdvals
  backward : svdvals_grad

- op : swiglu
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
     func : SwiGLUInferMeta
     spmd_rule : SwiGLUInferSpmd
  kernel :
     func : swiglu
  optional : y
  backward: swiglu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : swish
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
    param : [x]
  kernel :
    func : swish
  inplace: (x -> out)
  backward : swish_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : sync_batch_norm_
  args : (Tensor x, Tensor mean, Tensor variance, Tensor scale, Tensor bias, bool is_test, float momentum, float epsilon, str data_format, bool use_global_stats, bool trainable_statistics)
  output : Tensor(out), Tensor(mean_out), Tensor(variance_out), Tensor(saved_mean), Tensor(saved_variance), Tensor(reserve_space)
  infer_meta :
    func : BatchNormInferMeta
  kernel :
    func : sync_batch_norm
    data_type : x
  backward : sync_batch_norm_grad
  inplace : (mean -> mean_out), (variance -> variance_out)
  optional : reserve_space
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : sync_calc_stream
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
  kernel :
    func : sync_calc_stream
  inplace : (x -> out)
  traits : paddle::dialect::ForwardOnlyTrait

- op : take_along_axis
  args : (Tensor arr, Tensor indices, int axis)
  output : Tensor
  infer_meta :
    func : TakeAlongAxisInferMeta
    param : [arr, indices, axis]
    spmd_rule : TakeAlongAxisInferSpmd
  kernel :
    func : take_along_axis
    data_type : arr
  backward : take_along_axis_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : tan
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : tan
  inplace : (x -> out)
  backward : tan_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : tanh
  args : (Tensor x)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : tanh
  inplace : (x -> out)
  backward : tanh_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : tanh_shrink
  args : (Tensor x)
  output : Tensor
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : tanh_shrink
  backward : tanh_shrink_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : tdm_child
  args: (Tensor x, Tensor tree_info, int child_nums, DataType dtype = DataType::INT32)
  output: Tensor (child), Tensor (leaf_mask)
  infer_meta:
    func: TdmChildInferMeta
  kernel:
    func: tdm_child
    data_type: x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : tdm_sampler
  args: (Tensor x, Tensor travel, Tensor layer, bool output_positive=true, int[] neg_samples_num_list={}, int[] layer_offset={}, int seed = 0, int dtype=2)
  output: Tensor(out), Tensor(labels), Tensor(mask)
  infer_meta:
    func : TdmSamplerInferMeta
  kernel:
    func : tdm_sampler
    data_type : x
  optional : labels
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : temporal_shift
  args : (Tensor x, int seg_num, float shift_ratio = 0.25f, str data_format = "NCHW")
  output : Tensor(out)
  infer_meta :
    func : TemporalShiftInferMeta
  kernel :
    func : temporal_shift
    data_type : x
  backward : temporal_shift_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : thresholded_relu
  args : (Tensor x, float threshold = 1.0, float value = 0.0)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    param : [x]
    spmd_rule : ThresholdedReluInfoSpmd
  kernel :
    func : thresholded_relu
  inplace: (x -> out)
  backward : thresholded_relu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits: pir::UnaryElementWiseTrait

- op : top_p_sampling
  args : (Tensor x, Tensor ps, Tensor threshold, Tensor topp_seed, int64_t seed=-1, int k=0, str mode="truncate")
  output : Tensor (out), Tensor(ids), Tensor(topk_scores), Tensor(topk_ids)
  infer_meta :
    func : TopPSamplingInferMeta
  kernel :
    func : top_p_sampling
    data_type : x
  optional : threshold, topp_seed, topk_scores, topk_ids
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : topk
  args : (Tensor x, Scalar(int) k = 1, int axis = -1, bool largest = true, bool sorted = true)
  output : Tensor(out), Tensor(indices)
  infer_meta :
    func : TopKInferMeta
    spmd_rule: TopkInferSpmdDynamic
  kernel :
    func : topk
    data_type : x
  backward : topk_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : trace
  args : (Tensor x, int offset = 0, int axis1 = 0, int axis2 = 1)
  output : Tensor
  infer_meta :
    func : TraceInferMeta
  kernel :
    func : trace
  backward : trace_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : trans_layout
  args : (Tensor x, int[] perm)
  output : Tensor
  infer_meta :
    func : TransposeInferMeta
  kernel :
    func : transpose
  backward : trans_layout_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : transpose
  args : (Tensor x, int[] perm)
  output : Tensor(out)
  infer_meta :
    func : TransposeInferMeta
    spmd_rule: TransposeInferSpmd
  kernel :
    func : transpose
  inplace : (x -> out)
  backward : transpose_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : triangular_solve
  args : (Tensor x, Tensor y, bool upper=true, bool transpose=false, bool unitriangular=false)
  output : Tensor
  infer_meta :
    func : TriangularSolveInferMeta
  kernel :
    func : triangular_solve
    data_type : x
  backward : triangular_solve_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : tril
  args : (Tensor x, int diagonal=0)
  output : Tensor(out)
  infer_meta :
    func : TrilInferMeta
  kernel :
    func : tril
  inplace: (x -> out)
  backward : tril_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : tril_indices
  args : (int rows, int cols, int offset, DataType dtype, Place place={})
  output : Tensor(out)
  infer_meta :
    func : TrilIndicesInferMeta
    param : [rows, cols, offset, dtype]
  kernel :
    func : tril_indices
    param : [rows, cols, offset, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : trilinear_interp
  args : (Tensor x, Tensor out_size, Tensor[] size_tensor, Tensor scale_tensor, str data_format="NCHW", int out_d=0, int out_h=0, int out_w=0, double[] scale={}, str interp_method="bilinear", bool align_corners=true, int align_mode=1)
  output : Tensor(output)
  infer_meta :
    func : InterpolateInferMeta
  optional: out_size, size_tensor, scale_tensor
  kernel :
    func : trilinear_interp
    data_type : x
  backward : trilinear_interp_grad
  data_transform :
    skip_transform : out_size, size_tensor, scale_tensor
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : triu
  args : (Tensor x, int diagonal=0)
  output : Tensor(out)
  infer_meta :
    func : TriuInferMeta
    spmd_rule : TriuInferSpmd
  kernel :
    func : triu
  inplace: (x -> out)
  backward : triu_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : triu_indices
  args : (int row, int col, int offset, DataType dtype, Place place={})
  output : Tensor(out)
  infer_meta :
    func : TriuIndicesInferMeta
    param : [row, col, offset, dtype]
  kernel :
    func : triu_indices
    param : [row, col, offset, dtype]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : trunc
  args : (Tensor input)
  output : Tensor(out)
  infer_meta :
    func : UnchangedInferMeta
    spmd_rule : ElementwiseUnaryInferSpmd
  kernel :
    func : trunc
  inplace: (input -> out)
  backward : trunc_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : trunc_divide
  args : (Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : ElementwiseInferMeta
    spmd_rule : ElementwiseBinaryInferSpmd
  kernel :
    func : trunc_divide
  inplace: (x -> out)
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait, pir::BinaryElementWiseTrait

# python API: paddle.nn.initializer.TruncatedNormal
- op : truncated_gaussian_random
  args : (int[] shape, float mean, float std, int seed, float a, float b, DataType dtype=DataType::FLOAT32, Place place={})
  output : Tensor(out)
  infer_meta :
    func : TruncatedGaussianRandomInferMeta
    param : [shape, mean, std, seed, a, b, dtype]
  kernel :
    func : truncated_gaussian_random
    param : [shape, mean, std, seed, a, b, dtype]
    backend : place
    data_type : dtype
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : unbind
  args : (Tensor input, int axis = 0)
  output : Tensor[] {axis<0 ? input.dims()[input.dims().size()+axis]:input.dims()[axis]}
  infer_meta :
    func : UnbindInferMeta
    spmd_rule : UnbindInferSpmdDynamic
  kernel :
    func : unbind
  backward : unbind_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : unfold
  args : (Tensor x, int[] kernel_sizes, int[] strides, int[] paddings, int[] dilations)
  output : Tensor(out)
  infer_meta :
    func : UnfoldInferMeta
  kernel :
    func : unfold
  backward : unfold_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : uniform
  args : (IntArray shape,  DataType dtype,  Scalar min,  Scalar max,  int seed, Place place={})
  output : Tensor(out)
  infer_meta :
    func : UniformRandomInferMeta
    param: [shape, dtype]
  kernel :
    func : uniform
    param: [shape, dtype, min, max, seed]
    data_type : dtype
    backend : place
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait

- op : uniform_inplace
  args: (Tensor x, float min = -1.0, float max = 1.0, int seed = 0, int diag_num = 0, int diag_step = 0, float diag_val = 1.0)
  output: Tensor(out)
  infer_meta:
    func: UniformRandomInplaceInferMeta
  kernel:
    func: uniform_inplace
    data_type: x
  inplace: (x -> out)
  backward: uniform_inplace_grad
  traits : pir::SideEffectTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : uniform_random_batch_size_like
  args: (Tensor input, int[] shape, int input_dim_idx = 0, int output_dim_idx = 0,
    float min=-1.0f, float max=1.0f, int seed=0, int diag_num=0, int diag_step=0, float diag_val=1.0f, DataType dtype=DataType::FLOAT32)
  output: Tensor (out)
  infer_meta:
    func: UniformRandomBatchSizeLikeInferMeta
  kernel:
    func : uniform_random_batch_size_like {dense -> dense},
           uniform_random_batch_size_like_sr {selected_rows -> selected_rows}
    data_type: dtype
  no_need_buffer: input
  traits : pir::SideEffectTrait, paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : unique_consecutive
  args : (Tensor x, bool return_inverse = false, bool return_counts = false, int[] axis = {}, DataType dtype = DataType::FLOAT32)
  output : Tensor(out), Tensor(index), Tensor(counts)
  infer_meta :
      func : UniqueConsecutiveInferMeta
  kernel :
    func : unique_consecutive
    data_type : x
  optional : index, counts
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : unpool
  args: (Tensor x, Tensor indices, int[] ksize, int[] strides, int[] padding, IntArray output_size, str data_format)
  output: Tensor(out)
  infer_meta:
    func: UnpoolInferMeta
  kernel:
    func: unpool
    data_type: x
  backward: unpool_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : unpool3d
  args: (Tensor x, Tensor indices, int[] ksize, int[] strides={1,1,1}, int[] paddings={0,0,0}, int[] output_size={0,0,0}, str data_format="NCDHW")
  output: Tensor(out)
  infer_meta:
    func: Unpool3dInferMeta
  kernel:
    func: unpool3d
    data_type: x
  backward: unpool3d_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : unsqueeze
  args : (Tensor x, IntArray axis = {})
  output : Tensor(out)
  infer_meta :
    func : UnsqueezeInferMeta
    spmd_rule : UnsqueezeInferSpmd
  kernel :
    func : unsqueeze
    data_type : x
  inplace : (x -> out)
  view: (x -> out)
  backward : unsqueeze_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : unstack
  args : (Tensor x, int axis=0, int num=0)
  output : Tensor[](out){num}
  infer_meta :
    func : UnStackInferMeta
  kernel :
    func : unstack
  backward : unstack_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface, paddle::dialect::LayoutTransformationInterface

- op : update_loss_scaling_
  args : (Tensor[] x, Tensor found_infinite, Tensor prev_loss_scaling, Tensor in_good_steps, Tensor in_bad_steps, int incr_every_n_steps, int decr_every_n_nan_or_inf, float incr_ratio, float decr_ratio, Scalar stop_update=false)
  output : Tensor[](out){x.size()}, Tensor(loss_scaling), Tensor(out_good_steps), Tensor(out_bad_steps)
  infer_meta :
    func : UpdateLossScalingInferMeta
    param : [x, found_infinite, prev_loss_scaling, in_good_steps, in_bad_steps]
    spmd_rule : UpdateLossScalingSpmd
  kernel :
    func : update_loss_scaling
    data_type : x
  data_transform :
    skip_transform : found_infinite
  inplace : (x -> out), (prev_loss_scaling -> loss_scaling), (in_good_steps -> out_good_steps), (in_bad_steps -> out_bad_steps)
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : var
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false, bool unbiased=true, double correction=1)
  output : Tensor(out)
  infer_meta :
    func : VarInferMeta
  kernel :
    func : var
    data_type : x
  backward : var_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : variance
  args : (Tensor x, int64_t[] axis={}, bool keepdim=false)
  output : Tensor(out)
  infer_meta :
    func : ReduceInferMeta
  kernel :
    func : variance
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : view_dtype
  args : (Tensor input, DataType dtype)
  output : Tensor(out)
  infer_meta :
    func : StridedUnChangedInferMeta
    param : [input]
  kernel :
    func : view_dtype
    data_type : input
  backward : view_dtype_grad
  no_need_buffer : input
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : view_shape
  args : (Tensor input, int64_t[] dims = {})
  output : Tensor(out)
  infer_meta :
    func : ViewShapeInferMeta
  kernel :
    func : view_shape
  backward : view_shape_grad

- op : view_slice
  args : (Tensor input, int64_t begin_idx, int64_t end_idx)
  output : Tensor
  infer_meta :
    func : ViewSliceInferMeta
    spmd_rule : ViewSliceInferSpmd
  kernel :
    func : view_slice
  interfaces : paddle::dialect::InplaceTrait

- op : viterbi_decode
  args : (Tensor potentials, Tensor transition_params, Tensor lengths, bool include_bos_eos_tag = true)
  output : Tensor(scores), Tensor(path)
  infer_meta :
    func : ViterbiDecodeInferMeta
  kernel :
    func : viterbi_decode
    data_type : potentials
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : warpctc
  args : (Tensor logits, Tensor label, Tensor logits_length, Tensor labels_length, int blank = 0, bool norm_by_times = false)
  output :  Tensor(loss), Tensor(warpctcgrad)
  infer_meta :
    func : WarpctcInferMeta
  kernel :
    func : warpctc
    data_type : logits
  optional : logits_length, labels_length
  intermediate : warpctcgrad
  backward : warpctc_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : warprnnt
  args : (Tensor input, Tensor label, Tensor input_lengths, Tensor label_lengths, int blank = 0, float fastemit_lambda = 0.0)
  output :  Tensor(loss), Tensor(warprnntgrad)
  infer_meta :
    func : WarprnntInferMeta
  kernel :
    func : warprnnt
    data_type : input
  intermediate : warprnntgrad
  backward : warprnnt_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : weight_dequantize
  args : (Tensor x, Tensor scale, str algo = "weight_only_int8", int group_size = -1)
  output : Tensor(out)
  infer_meta :
    func : WeightDequantizeInferMeta
  kernel :
    func : weight_dequantize
    data_type : scale
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : weight_only_linear
  args : (Tensor x, Tensor weight, Tensor bias, Tensor weight_scale, str weight_dtype, int arch = 80, int group_size = -1)
  output : Tensor(out)
  infer_meta :
    func : WeightOnlyLinearInferMeta
  kernel :
    func : weight_only_linear
    data_type : x
  optional : bias
  backward : weight_only_linear_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : weight_quantize
  args : (Tensor x, str algo = "weight_only_int8", int arch = 80, int group_size = -1)
  output : Tensor(out), Tensor(scale)
  infer_meta :
    func : WeightQuantizeInferMeta
  kernel :
    func : weight_quantize
    data_type : x
    backend : x
  traits : paddle::dialect::ForwardOnlyTrait
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : weighted_sample_neighbors
  args : (Tensor row, Tensor colptr, Tensor edge_weight, Tensor input_nodes, Tensor eids, int sample_size, bool return_eids)
  output : Tensor(out_neighbors), Tensor(out_count), Tensor(out_eids)
  infer_meta :
    func : WeightedSampleNeighborsInferMeta
  kernel :
    func : weighted_sample_neighbors
  optional : eids
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : where
  args : (Tensor condition, Tensor x, Tensor y)
  output : Tensor(out)
  infer_meta :
    func : WhereInferMeta
    spmd_rule: WhereInferSpmd
  kernel :
    func : where
  inplace : (x -> out)
  backward : where_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : yolo_box
  args : (Tensor x, Tensor img_size, int[] anchors={}, int class_num = 1, float conf_thresh = 0.01, int downsample_ratio = 32, bool clip_bbox = true, float scale_x_y=1.0, bool iou_aware=false, float iou_aware_factor=0.5)
  output : Tensor(boxes), Tensor(scores)
  infer_meta :
    func : YoloBoxInferMeta
  kernel :
    func : yolo_box
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : yolo_box_head
  args : (Tensor x, int[] anchors, int class_num)
  output : Tensor(out)
  infer_meta :
    func : YoloBoxHeadInferMeta
  kernel :
    func : yolo_box_head
    data_type : x
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : yolo_box_post
  args : (Tensor boxes0, Tensor boxes1, Tensor boxes2, Tensor image_shape, Tensor image_scale, int[] anchors0, int[] anchors1, int[] anchors2, int class_num, float conf_thresh, int downsample_ratio0, int downsample_ratio1, int downsample_ratio2, bool clip_bbox, float scale_x_y, float nms_threshold)
  output : Tensor(out), Tensor(nms_rois_num)
  infer_meta :
    func : YoloBoxPostInferMeta
  kernel :
    func : yolo_box_post
    data_type : boxes0
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op : yolo_loss
  args : (Tensor x, Tensor gt_box, Tensor gt_label, Tensor gt_score, int[] anchors={}, int[] anchor_mask={}, int class_num =1 , float ignore_thresh=0.7, int downsample_ratio=32, bool use_label_smooth=true, float scale_x_y=1.0)
  output : Tensor(loss), Tensor(objectness_mask), Tensor(gt_match_mask)
  infer_meta :
    func : YoloLossInferMeta
  kernel :
    func : yolo_loss
    data_type : x
  optional : gt_score
  intermediate : objectness_mask, gt_match_mask
  backward : yolo_loss_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface

- op : zeros
  args : (IntArray shape, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
  output : Tensor(out)
  invoke : full(shape, 0, dtype, place)

- op : zeros_like
  args : (Tensor x, DataType dtype=DataType::UNDEFINED, Place place = {})
  output : Tensor(out)
  invoke : full_like(x, 0, dtype, place)

- op: batched_gemm
  args: (Tensor lhs, Tensor rhs, int64_t[] batch_sizes, bool trans_lhs, bool trans_rhs)
  output: Tensor(output)
  infer_meta:
    func: BatchedGemmInferMeta
  kernel:
   func: batched_gemm
   data_type: lhs

- op: chunk_eval
  args: (Tensor inference, Tensor label, Tensor seq_length, int num_chunk_types, str
    chunk_scheme = "IOB", int[] excluded_chunk_types = {})
  output: Tensor (precision), Tensor (recall), Tensor (f1_score), Tensor (num_infer_chunks),
    Tensor (num_label_chunks), Tensor (num_correct_chunks)
  infer_meta:
    func: ChunkEvalInferMeta
  kernel:
    func: chunk_eval
    data_type: DataType::FLOAT32
  optional: seq_length
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op: fast_ln
  args: (Tensor x, Tensor scale, Tensor bias, float epsilon)
  output: Tensor(y), Tensor(mean), Tensor(invvar)
  infer_meta:
    func: FastLayerNormInfermeta
  kernel:
    func: fast_ln
    data_type: scale
  backward: fast_ln_grad

- op: fast_rms_norm
  args: (Tensor x, Tensor scale, float epsilon)
  output: Tensor(y), Tensor(invvar)
  infer_meta:
    func: FastRMSNormInfermeta
  kernel:
    func: fast_rms_norm
    data_type: scale
  backward: fast_rms_norm_grad

- op: fp8_gemm_blockwise_
  args: (Tensor A, Tensor A_scale, Tensor B, Tensor B_scale, Tensor input_result, Tensor bias, Tensor pre_gelu, Tensor workspace, bool transa, bool transb, bool grad, bool accumulate, bool use_split_accumulator, int math_sm_count, bool is_A_1d_scaled, bool is_B_1d_scaled)
  output: Tensor (output), Tensor (pre_gelu_out), Tensor (workspace_out)
  inplace: (input_result -> output), (pre_gelu -> pre_gelu_out), (workspace -> workspace_out)
  infer_meta:
    func: Fp8GemmBlockwiseInferMeta
  kernel:
    func: fp8_gemm_blockwise

- op: fp8_quant_blockwise
  args: (Tensor x, float epsilon, bool using_1x128_vec_quant, bool input_transpose, bool output_scale_transpose, bool return_transpose_only, bool using_e5m2, bool using_pow2_scale, bool using_ue8m0_scale)
  output: Tensor(out), Tensor(scale), Tensor(out_transposed), Tensor(scale_transposed)
  infer_meta:
    func: Fp8QuantBlockwiseInferMeta
  kernel:
    func: fp8_quant_blockwise
    data_type: x

- op: fused_rms_norm_ext
  args: (Tensor x, Tensor scale, float epsilon)
  output: Tensor(y), Tensor(invvar)
  infer_meta:
    func: FusedRMSNormInferMeta
  kernel:
    func: fused_rms_norm_ext
    data_type: x
  backward: fused_rms_norm_ext_grad

- op: int_bincount
  args: (Tensor x, int64_t low, int64_t high, int64_t dtype)
  output: Tensor(out)
  infer_meta:
    func: IntBincountInferMeta
  kernel:
    func: int_bincount
    data_type: x

- op: number_count
  args: (Tensor numbers, int upper_range)
  output: Tensor(out)
  infer_meta:
     func: NumberCountInferMeta
  kernel:
     func: number_count
     data_type: numbers
  interfaces : paddle::dialect::InferSymbolicShapeInterface
  traits : paddle::dialect::ForwardOnlyTrait

- op: rms_norm
  args: (Tensor x, Tensor scale, int64_t[] normalized_shape={}, double epsilon= 1e-5)
  output: Tensor(y), Tensor(invvar)
  infer_meta:
    func: RmsNormInferMeta
  kernel:
    func: rms_norm
    data_type: x
  optional : scale
  backward: rms_norm_grad
  interfaces : paddle::dialect::InferSymbolicShapeInterface