Function mindspore::ops::ArgMaxInfer

• Defined in File arg_max.h

Function Documentation

AbstractBasePtr mindspore::ops::ArgMaxInfer(const abstract::AnalysisEnginePtr&, const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args)