Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h141 operator_property::OperatorProperty lstm_property; in matchAndRewrite() local
143 if (failed(GetLstmProperty(op, &lstm_variant, &lstm_property))) { in matchAndRewrite()
146 if (lstm_property.restrict_scale.size() != 1) { in matchAndRewrite()
149 << lstm_property.restrict_scale.size() in matchAndRewrite()
155 const std::vector<int>& tensors = lstm_property.restrict_scale[0]; in matchAndRewrite()
412 operator_property::OperatorProperty lstm_property; in matchAndRewrite() local
413 if (failed(GetLstmProperty(op, &lstm_variant, &lstm_property))) { in matchAndRewrite()
417 if (failed(processIntermediates(op, lstm_variant, lstm_property)) || in matchAndRewrite()
419 op, lstm_variant, lstm_property, rewriter))) { in matchAndRewrite()
429 const operator_property::OperatorProperty& lstm_property) const { in processIntermediates()
[all …]