Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize.cc210 auto q_op = in SetInputNodesQuantizationParams() local
213 q_op.getResult()); in SetInputNodesQuantizationParams()
215 q_op.setOperand(arg); in SetInputNodesQuantizationParams()
292 func.walk([&](quant::QuantizeCastOp q_op) { in SanityCheckAndAdjustment() argument
295 q_op.getOperand().getDefiningOp()); in SanityCheckAndAdjustment()
321 << " quantizer's output has another quantizer (" << q_op.getLoc() in SanityCheckAndAdjustment()
Doptimize.cc557 auto q_op = in matchAndRewrite() local
559 if (!q_op) return failure(); in matchAndRewrite()
560 filter = q_op.input(); in matchAndRewrite()
573 rewriter.setInsertionPoint(q_op); in matchAndRewrite()
591 auto new_qtype = RescaleQtype(q_op.qtype(), gamma_cst); in matchAndRewrite()
593 rewriter.replaceOpWithNewOp<TFL::QuantizeOp>(q_op, new_qtype.getValue(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconvert_type.cc159 mlir::TFL::QuantizeOp q_op = in NotFromQuantOpOrSameQuantType() local
161 if (!q_op) return true; in NotFromQuantOpOrSameQuantType()
165 auto stripped_src_qtype = GetShapeStrippedType(q_op.qtypeAttr()); in NotFromQuantOpOrSameQuantType()