- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 85 for matchAndRewrite (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->erase(); }); } class ConvertGraphOp : public OpConversionPattern<tfg::GraphOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( tfg::GraphOp graph, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const final { Location loc = graph.getLoc(); // To keep the import-as-graph logic taken by TFG, we create `void func()`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
void runOnOperation() override; }; class HandleOptionalFrom : public OpRewritePattern<TF::OptionalFromValueOp> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::OptionalFromValueOp op, PatternRewriter& rewriter) const override { Value value = nullptr; for (auto v : op.getComponents()) { value = v; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
public: explicit MergeConsecutiveQuantizeCast(MLIRContext* context) : OpRewritePattern<quantfork::QuantizeCastOp>(context) {} private: LogicalResult matchAndRewrite(quantfork::QuantizeCastOp q_op, PatternRewriter& rewriter) const override { auto preceding_qcast = q_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/post_quantize.cc
: public OpRewritePattern<quantfork::DequantizeCastOp> { public: explicit RemoveVolatileQdqPattern(MLIRContext* context) : OpRewritePattern<quantfork::DequantizeCastOp>(context) {} LogicalResult matchAndRewrite(quantfork::DequantizeCastOp op, PatternRewriter& rewriter) const override { auto input_op = op.getArg().getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
} LogicalResult ConvertTFBatchMatMulOp::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const { if (ConvertTFBatchMatMulOp2TFLFullyConnectedOp<TF::BatchMatMulOp>(op, rewriter)) return success(); return failure(); } LogicalResult ConvertTFBatchMatMulV2Op::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
class LegalizeHashTableOpPattern : public OpRewritePattern<TF::HashTableV2Op> { public: using OpRewritePattern<TF::HashTableV2Op>::OpRewritePattern; LogicalResult matchAndRewrite(TF::HashTableV2Op hashtable_op, PatternRewriter& rewriter) const override { auto output_type = RankedTensorType::get( {1}, TF::ResourceType::get(rewriter.getContext()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
// on the tensor proto. class ConvertUniformQuantizedDotHybridOp : public OpConversionPattern<TF::UniformQuantizedDotHybridOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::UniformQuantizedDotHybridOp op, TF::UniformQuantizedDotHybridOpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { // Uniform Quantized type for the rhs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
LogicalResult EnsureBiasForConv2d::matchAndRewrite( TFL::Conv2DOp conv_op, PatternRewriter& rewriter) const { return EnsureBias(conv_op, 2, rewriter); } // ================== slice ============================ // If a slice op has < 4d dimension, will pad it to 4d. LogicalResult PadSlice::matchAndRewrite(TFL::SliceOp slice_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc
// the input is sign-extended two's complement. class CheckRangeAndConvertI8ToI4 : public OpRewritePattern<arith::ConstantOp> { public: using OpRewritePattern<arith::ConstantOp>::OpRewritePattern; LogicalResult matchAndRewrite(arith::ConstantOp op, PatternRewriter &rewriter) const override { auto const_type = mlir::dyn_cast<ShapedType>(op.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc
LogicalResult matchAndRewrite(QuantizeCastOp qbarrier, PatternRewriter &rewriter) const override; }; } // namespace /// Matches a [constant] -> [qbarrier] where the qbarrier results type is /// quantized and the operand type is quantizable. LogicalResult QuantizedConstRewrite::matchAndRewrite( QuantizeCastOp qbarrier, PatternRewriter &rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0)