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

QuantizeWeightsOnly quantizes tensors for weights. More...

#include <QuantizeWeightsOnly.h>

Collaboration diagram for luci::QuantizeWeightsOnly:

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
 

Detailed Description

QuantizeWeightsOnly quantizes tensors for weights.

Find min/max values on the fly and then quantize

Definition at line 30 of file QuantizeWeightsOnly.h.

Constructor & Destructor Documentation

◆ QuantizeWeightsOnly()

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

Definition at line 32 of file QuantizeWeightsOnly.h.

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

Field Documentation

◆ granularity

QuantizationGranularity luci::QuantizeWeightsOnly::granularity

Definition at line 39 of file QuantizeWeightsOnly.h.

◆ input_type

loco::DataType luci::QuantizeWeightsOnly::input_type

Definition at line 37 of file QuantizeWeightsOnly.h.

◆ output_type

loco::DataType luci::QuantizeWeightsOnly::output_type

Definition at line 38 of file QuantizeWeightsOnly.h.


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