- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 37 for matchAndRewrite (0.36 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/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/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/legalize_tensorlist.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc" struct ConvertTensorListPopBack : public OpRewritePattern<TF::TensorListPopBackOp> { using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::TensorListPopBackOp op, PatternRewriter& rewriter) const override { // It is currently not possible to easily pack the output of a multi-result
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
struct RemoveVolatileOps : public OpRewritePattern<DequantizeOp> { explicit RemoveVolatileOps(MLIRContext* context) : OpRewritePattern<DequantizeOp>(context, 1) {} LogicalResult matchAndRewrite(DequantizeOp op, PatternRewriter& rewriter) const override { auto input_op = op.getInput().getDefiningOp(); if (auto q = llvm::dyn_cast_or_null<QuantizeOp>(input_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
class TFQuantTypePattern : public ConversionPattern { public: TFQuantTypePattern(MLIRContext *ctx, TypeConverter &converter) : ConversionPattern(converter, MatchAnyOpTypeTag(), 1, ctx) {} LogicalResult matchAndRewrite( Operation *op, ArrayRef<Value> operands, ConversionPatternRewriter &rewriter) const override { // This pattern only handle non-UQ, non-const ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
template <typename SourceOp> class PrepareLstmOutputScale : public OpRewritePattern<SourceOp> { public: explicit PrepareLstmOutputScale(MLIRContext* context) : OpRewritePattern<SourceOp>(context) {} LogicalResult matchAndRewrite(SourceOp op, PatternRewriter& rewriter) const override { operator_property::OpVariant lstm_variant; operator_property::OperatorProperty lstm_property;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
} class UnfuseBatchNormTrainingPattern : public OpRewritePattern<mhlo::BatchNormTrainingOp> { public: using OpRewritePattern<mhlo::BatchNormTrainingOp>::OpRewritePattern; LogicalResult matchAndRewrite(mhlo::BatchNormTrainingOp bn_op, PatternRewriter &rewriter) const override { auto inputs = bn_op.getOperand(); auto input_type = mlir::dyn_cast<RankedTensorType>(inputs.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0)