Skip to content
Snippets Groups Projects

Add TopK operator

Merged Olivier BICHLER requested to merge topk into dev
2 files
+ 9
0
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -94,6 +94,7 @@ public:
bool forwardDims(bool allowDataDependency = false) override final;
void setBackend(const std::string& name, DeviceIdx_t device = 0) override;
void setDataType(const DataType& dataType) const override final;
std::set<std::string> getAvailableBackends() const override;
inline std::shared_ptr<Attributes> attributes() const override { return mAttributes; }
Loading