Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CanFuseConvOrDepthwiseConv (0.51 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td

      defm : FuseActFnIntoPoolOpPat<!cast<Op>(actFnPair[0]), !cast<ConstantStrAttr>(actFnPair[1])>;
    }
    
    class CanFuseConvOrDepthwiseConv<string is_depthwise> : Constraint<
      CPred<"TFL::CanFuseConvOrDepthwiseConv($0, $1, " # is_depthwise # ")">>;
    
    // If we see a binary op (add, sub) op adding a constant value to a convolution
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 66.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/optimize.cc

      if (filter_shape.empty() ||
          (is_depthwise ? filter_shape.back() != elements_depth
                        : filter_shape[0] != elements_depth))
        return false;
      return true;
    }
    
    bool CanFuseConvOrDepthwiseConv(Value filter, Attribute val,
                                    bool is_depthwise) {
      const auto elements = mlir::dyn_cast<DenseElementsAttr>(val);
      if (!elements) {
        return false;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top