Function mindspore::ops::TopKInfer

• Defined in File topk.h

Function Documentation

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