Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for MulOp (0.04 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize.cc

    // TODO(b/136285429): Move to tablegen when variadic is supported
    struct FuseFullyConnectedAndMul : public OpRewritePattern<TFL::MulOp> {
      using OpRewritePattern<TFL::MulOp>::OpRewritePattern;
    
      LogicalResult matchAndRewrite(TFL::MulOp mul_op,
                                    PatternRewriter &rewriter) const override {
        // If we are broadcasting on the lhs then don't fold the multiply as it
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

    }
    
    //===----------------------------------------------------------------------===//
    // MulOp
    //===----------------------------------------------------------------------===//
    
    OpFoldResult MulOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      return IdentityArithmeticOpFolder<MulOp>(*this, operands);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

            }
          }
        }
      }
      return success();
    }
    
    //===----------------------------------------------------------------------===//
    // MulOp
    //===----------------------------------------------------------------------===//
    
    OpFoldResult MulOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      // TODO(b/142478136): Handle fused ops.
      if (getFusedActivationFunction() != "NONE") return {};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        if (!mlir::isa<RankedTensorType>(reduce_op.getType(0))) return failure();
        return success();
      }
    };
    
    class ConvertReduceOpToTfProd
        : public ConvertReduceOpToTfOp<mhlo::MulOp, TF::ProdOp, TF::MulOp> {
     public:
      using ConvertReduceOpToTfOp::ConvertReduceOpToTfOp;
    
      LogicalResult MatchInitValue(Value init_value) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        Pure,
        Commutative,
        QuantizableResult,
        BinaryOpSameElementTypeConstraint,
        TFL_RuntimePredOpTrait<"Operands do not have valid shapes",
          CPred<"TFL::VerifyMulOpShapeConstraints(llvm::cast<MulOp>($_op))">>,
        DeclareOpInterfaceMethods<TFL_ArithmeticCount>]> {
      let summary = "Multiplication operator";
    
      let description = [{
        Element-wise multiplication operation.
      }];
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top