- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for kVolatileOpAttrName (0.21 sec)
-
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
dq.getArg()); dq.getResult().replaceAllUsesWith(dcast); if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); } dq.erase(); } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) { auto out_type = q.getResult().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/post_quantize.cc
PatternRewriter& rewriter) const override { auto input_op = op.getArg().getDefiningOp(); if (auto q = llvm::dyn_cast_or_null<quantfork::QuantizeCastOp>(input_op)) { if (!q->getAttr(kVolatileOpAttrName)) return failure(); // If the quantize op is a requantize op, it is being used in other scale // adjustments and should be kept. Instead, move dequantize op before the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/post_quantize.cc
PatternRewriter& rewriter) const override { auto input_op = op.getArg().getDefiningOp(); if (auto q = llvm::dyn_cast_or_null<quantfork::QuantizeCastOp>(input_op)) { if (!q->getAttr(kVolatileOpAttrName)) return failure(); if (remove_volatile_ops_type == kPreserveInputsAndOutputs) { // Don't remove leading and trailing QDQ for PTQ workflow, so the io // modifying lib can work correctly.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
if (!preceding_qcast) return failure(); auto new_qcast = rewriter.create<quantfork::QuantizeCastOp>( q_op.getLoc(), q_op.getType(), preceding_qcast.getArg()); new_qcast->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); q_op->replaceAllUsesWith(new_qcast); return success(); } }; class ConvertTFConstOpToArithConstOp : public OpRewritePattern<TF::ConstOp> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0)