|
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.