Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h87 LstmOp op, operator_property::OpVariant* lstm_variant, in GetLstmProperty() argument
90 lstm_variant->op_code = tflite::BuiltinOperator_LSTM; in GetLstmProperty()
92 lstm_variant->op_code = in GetLstmProperty()
98 lstm_variant->use_projection = in GetLstmProperty()
100 lstm_variant->use_peephole = in GetLstmProperty()
102 lstm_variant->use_layer_norm = in GetLstmProperty()
105 *op_property = operator_property::GetOperatorProperty(*lstm_variant); in GetLstmProperty()
140 operator_property::OpVariant lstm_variant; in matchAndRewrite() local
143 if (failed(GetLstmProperty(op, &lstm_variant, &lstm_property))) { in matchAndRewrite()
411 operator_property::OpVariant lstm_variant; in matchAndRewrite() local
[all …]