- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for MulOp (0.04 sec)
-
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) -
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) -
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) -
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) -
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)