Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for FuseMulOrDivWithConv2dOrDepthwiseConv2d (0.44 sec)

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

    // into the convolution op by constant folding
    // the filter/bias and the div/mul op's constant operand.
    // The following pattern restricts to float constant values for now.
    
    multiclass FuseMulOrDivWithConv2dOrDepthwiseConv2d<Op BinaryOp> {
      def FuseMulOrDivWithDepthwiseConv#BinaryOp : Pat<
        (BinaryOp (TFL_DepthwiseConv2DOp:$output $input,
                    (Arith_ConstantOp FloatElementsAttr:$filter),
    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/ir/tfl_ops.cc

      if (getFusedActivationFunction() != "NONE") return {};
    
      // This function is performance critical for op fusion patterns, e.g.
      // FuseBinaryOpToPrecedingAffine and FuseMulOrDivWithConv2dOrDepthwiseConv2d.
      // So a few specializations are provided to evaluate the math operation
      // more efficiently.
    
      // Specialization for f32 type.
      if (getType().cast<ShapedType>().getElementType().isF32()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top