Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dtfl_to_std.cc51 if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { in ConvertMlirQuantOpsToTFLQuantOps()
52 dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); in ConvertMlirQuantOpsToTFLQuantOps()
60 if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { in ConvertMlirQuantOpsToTFLQuantOps()
61 qcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); in ConvertMlirQuantOpsToTFLQuantOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc133 if (!q->getAttr(mlir::quant::kVolatileOpAttrName)) return failure(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.h50 constexpr char kVolatileOpAttrName[] = "volatile"; variable
156 q->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); in matchAndRewrite()
Dquantization_driver.cc540 quantize->setAttr(kVolatileOpAttrName, builder_.getUnitAttr()); in QuantizeValue()