SimaArgMax.h File

SimaArgMax Node — postprocess argmax over a tensor (classification). More...

Included Headers

#include "builder/Node.h" #include "builder/OutputSpec.h" #include <memory> #include <string> #include <vector>

Namespaces Index

namespace	simaai
namespace	neat
namespace	nodes

Classes Index

struct	SimaArgMaxOptions
	Construction options for a SimaArgMax Node. More...
class	SimaArgMax
	EV74 postprocess Node that emits the top-class index from a classification tensor. More...

Description

SimaArgMax Node — postprocess argmax over a tensor (classification).

Runs on the EV74. Reads a classification tensor (after dequant/cast) and emits the top class index per sample. Place after Detess/DetessDequant/DetessCast at the tail of a classification pipeline.

---

Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.8.