Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetUserIfOnlyOne (0.42 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc

                                    PatternRewriter& rewriter) const override {
        if (!MatchERF(op)) return failure();
        // `add 1`
        auto* erf_user = GetUserIfOnlyOne(op);
        if (!erf_user) return failure();
    
        // `mul`
        auto* erf_user_user = GetUserIfOnlyOne(erf_user);
        if (!erf_user_user) return failure();
    
        // `mul 1/sqrt(2)`
        auto* erf_input = GetInputOpWithOneUse(op, 0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 9.6K bytes
    - Viewed (0)
Back to top