experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add missing explicit dependent dialect for PrepareQuantizePass (NFC)

Browse Source 
PiperOrigin-RevId: 328503594
Change-Id: I7c3dcc3cab0b2db37dcf24608e10e23d252f9fe6
This commit is contained in:
Mehdi Amini 2020-08-26 03:55:48 -07:00 committed by TensorFlower Gardener
parent ed6d0f3e80
commit b005ebf734
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc
View File

@ -70,7 +70,8 @@ namespace {
class PrepareQuantizePass class PrepareQuantizePass
: public PassWrapper<PrepareQuantizePass, FunctionPass> { : public PassWrapper<PrepareQuantizePass, FunctionPass> {
void getDependentDialects(DialectRegistry& registry) const override { void getDependentDialects(DialectRegistry& registry) const override {
registry.insert<TFL::TensorFlowLiteDialect>(); registry.insert<TFL::TensorFlowLiteDialect,
::mlir::quant::QuantizationDialect>();
} }
public: public: