ONE - On-device Neural Engine
|
QuantizeWeightsOnly quantizes tensors for weights. More...
#include <QuantizeWeightsOnly.h>
Public Member Functions | |
QuantizeWeightsOnly (loco::DataType input, loco::DataType output, QuantizationGranularity gr) | |
Public Member Functions inherited from luci::CircleNodeMutableVisitor< void > | |
virtual | ~CircleNodeMutableVisitor ()=default |
Public Member Functions inherited from luci::CircleNodeMutableVisitorBase< T > | |
virtual | ~CircleNodeMutableVisitorBase ()=default |
Data Fields | |
loco::DataType | input_type |
loco::DataType | output_type |
QuantizationGranularity | granularity |
QuantizeWeightsOnly quantizes tensors for weights.
Find min/max values on the fly and then quantize
Definition at line 30 of file QuantizeWeightsOnly.h.
|
inline |
Definition at line 32 of file QuantizeWeightsOnly.h.
QuantizationGranularity luci::QuantizeWeightsOnly::granularity |
Definition at line 39 of file QuantizeWeightsOnly.h.
loco::DataType luci::QuantizeWeightsOnly::input_type |
Definition at line 37 of file QuantizeWeightsOnly.h.
loco::DataType luci::QuantizeWeightsOnly::output_type |
Definition at line 38 of file QuantizeWeightsOnly.h.