Home
last modified time | relevance | path

Searched refs:patterns_2 (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize.cc381 OwningRewritePatternList patterns_2; in runOnFunction() local
383 patterns_2.insert<quant::ConvertUnsignedToSigned<quant::QuantizeCastOp>>( in runOnFunction()
387 patterns_2.insert<PrepareQuantStats>(bit_width, false, true, in runOnFunction()
392 patterns_2.insert<PrepareQuantStats>(bit_width, false, false, in runOnFunction()
397 patterns_2.insert<ConvertLstmStatsToQDQs<LSTMOp>>(ctx, quant_specs_); in runOnFunction()
398 patterns_2.insert<ConvertLstmStatsToQDQs<UnidirectionalSequenceLSTMOp>>( in runOnFunction()
400 patterns_2.insert<ConvertSvdfStatsToQDQs>(ctx, quant_specs_); in runOnFunction()
402 (void)applyPatternsAndFoldGreedily(func, std::move(patterns_2)); in runOnFunction()