Other Operators

We have added Python API support to the following operators. The parameters match those in the same-named OPs in atb/infer_op_params.h. Parameter names use snake_case. For other calling methods, see OpParam.

  • AllGatherParam
  • AllReduceParam
  • AllToAllParam
  • AsStridedParam
  • BroadcastParam
  • ConcatParam
  • CumsumParam
  • DynamicNTKParam
  • FaUpdateParam
  • FillParam
  • GatingParam
  • LinearParallelParam
  • LinearSparseParam
  • MultinomialParam
  • PagedAttentionParam
  • ReshapeAndCacheParam
  • RazorFusionAttentionParam
  • ReduceScatterParam
  • ReduceScatterVParam
  • RelayAttentionParam
  • SliceParam
  • TopkToppSamplingParam
  • TransposeParam