Skip to content
Snippets Groups Projects

Add TopK operator

Merged Olivier BICHLER requested to merge topk into dev
4 files
+ 51
0
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -108,6 +108,14 @@ public:
static const std::vector<std::string> getOutputsName(){
return {"values", "indices"};
}
/**
* @brief Retrieves the names of the attributes for the operator.
* @return A vector containing the attributes name.
*/
static constexpr const char* const* attributesName(){
return EnumStrings<Aidge::TopKAttr>::data;
}
};
std::shared_ptr<Node> TopK(const std::string& name = "");
Loading