ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::QuantizeWeights Struct Referencefinal

QuantizeWeights quantizes tensors for weights. More...

#include <QuantizeWeights.h>

Collaboration diagram for luci::QuantizeWeights:

Public Member Functions

 QuantizeWeights (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
 

Detailed Description

QuantizeWeights quantizes tensors for weights.

Find min/max values on the fly and then quantize

Definition at line 30 of file QuantizeWeights.h.

Constructor & Destructor Documentation

◆ QuantizeWeights()

luci::QuantizeWeights::QuantizeWeights ( loco::DataType  input,
loco::DataType  output,
QuantizationGranularity  gr 
)
inline

Definition at line 32 of file QuantizeWeights.h.

33 : input_type(input), output_type(output), granularity(gr)
34 {
35 }
QuantizationGranularity granularity
loco::DataType input_type
loco::DataType output_type

Field Documentation

◆ granularity

QuantizationGranularity luci::QuantizeWeights::granularity

Definition at line 39 of file QuantizeWeights.h.

◆ input_type

loco::DataType luci::QuantizeWeights::input_type

Definition at line 37 of file QuantizeWeights.h.

◆ output_type

loco::DataType luci::QuantizeWeights::output_type

Definition at line 38 of file QuantizeWeights.h.


The documentation for this struct was generated from the following files: