Home
last modified time | relevance | path

Searched refs:TopKInfer (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dtopk.h49 AbstractBasePtr TopKInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
Dtopk.cc31 AbstractBasePtr TopKInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in TopKInfer() function