- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ConvolutionOp (0.13 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
// * Dst dimension numbers: [b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f] class RewriteNchwConvolutionToNhwc : public OpRewritePattern<mlir::stablehlo::ConvolutionOp> { public: using OpRewritePattern<mlir::stablehlo::ConvolutionOp>::OpRewritePattern; LogicalResult match(mlir::stablehlo::ConvolutionOp op) const override { // Handles 2D convolutions only. if (!HasRankOf(op.getOperand(0), /*rank=*/4) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
LogicalResult matchAndRewrite(mhlo::MulOp mul_op, PatternRewriter &rewriter) const override { // Variables for capturing values and attributes used while creating ops. mhlo::ConvolutionOp conv_op; Operation *bcast_or_const_op; shape::ShapeOfOp shape_of_op; mhlo::ConstantOp filter; mhlo::ConstantOp multiplier; mlir::ElementsAttr filter_value, mul_value;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/optimize.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" include "mhlo/IR/hlo_ops.td" include "stablehlo/dialect/ChloOps.td" def IsDefinedByConvOrDotGeneralOp : Constraint<Or<[ CPred<"$0.getDefiningOp<mhlo::ConvolutionOp>()">, CPred<"$0.getDefiningOp<mhlo::DotGeneralOp>()">]>>; def IsNull : Constraint<CPred<"!$0">>; // This pattern optimizes: // conv/dot_general + a + b -> conv/dot_general + (a + b)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 24 02:26:47 UTC 2024 - 2K bytes - Viewed (0)