Searched refs:op_variant (Results 1 – 3 of 3) sorted by relevance
27 OpVariant op_variant; in GetOperatorVariant() local30 op_variant.op_code = in GetOperatorVariant()32 if (op_variant.op_code == BuiltinOperator_LSTM || in GetOperatorVariant()33 op_variant.op_code == BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM) { in GetOperatorVariant()36 op_variant.is_quantizable = false; in GetOperatorVariant()37 return op_variant; in GetOperatorVariant()42 op_variant.use_projection = op->inputs[projection_weights_index] != -1; in GetOperatorVariant()43 op_variant.use_peephole = op->inputs[cell_to_output_weight_index] != -1; in GetOperatorVariant()45 op_variant.use_layer_norm = false; in GetOperatorVariant()47 op_variant.use_layer_norm = in GetOperatorVariant()[all …]
145 OperatorProperty GetOperatorProperty(OpVariant op_variant);
235 SourceOp op, const operator_property::OpVariant& op_variant, in processInputs()562 operator_property::OpVariant op_variant; in matchAndRewrite() local563 op_variant.op_code = tflite::BuiltinOperator_SVDF; in matchAndRewrite()564 auto op_property = operator_property::GetOperatorProperty(op_variant); in matchAndRewrite()566 op, op_variant, op_property, rewriter); in matchAndRewrite()