My Project
|
#include <Statistics.h>
Public Member Functions | |
AttributeTensorStatistics (Attribute attr) | |
bool | get (TensorAxisStatistics &stats) const override |
![]() | |
virtual | ~AbstractTensorStatistics ()=default |
virtual bool | supportsPerAxis () const |
virtual unsigned | getAxisCount () const |
Count of axes supported in a per-axis query. More... | |
virtual bool | getForAxis (unsigned axis, TensorAxisStatistics &stats) const |
Wraps an MLIR Attribute and returns statistics about it. It is expected that the attribute be one of: FloatAttr (scalar) DenseFPElementsAttr OpaqueElementsAttr (with Float based type) SparseElementAttr (with Float based type)
|
inline |
|
overridevirtual |
Gets statistics across the whole tensor. Returns true if statistics are valid and were populated.
Reimplemented from mlir::quantizer::AbstractTensorStatistics.