File arg_max_fusion.h

↰ Parent directory (include/converter/include/ops/fusion)

Definition (include/converter/include/ops/fusion/arg_max_fusion.h)

• Program Listing for File arg_max_fusion.h

Includes

• memory

• ops/arg_max.h (File arg_max.h)

• vector (File shape_vector.h)

Namespaces

• mindspore

• mindspore::ops

Classes

• Class ArgMaxFusion

Functions

• Function mindspore::ops::ArgMaxFusionInfer

Typedefs

• Typedef mindspore::ops::PrimArgMaxFusion

Variables

• Variable mindspore::ops::kNameArgMaxFusion