- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 100 for matchAndRewrite (0.39 sec)
-
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) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
}; class RemoveTpuOp : public RewritePattern { public: explicit RemoveTpuOp(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} private: LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override { // Remove `_tpu_replicate` attributes on each operation first. if (op->hasAttr(tensorflow::kTPUReplicateAttr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/post_quantize.cc
struct RemoveVolatileOps : public OpRewritePattern<quantfork::DequantizeCastOp> { explicit RemoveVolatileOps(MLIRContext* context) : OpRewritePattern<quantfork::DequantizeCastOp>(context, 1) {} 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 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
namespace mlir { namespace TFR { namespace { class ConvertConstToTensorConst : public OpRewritePattern<ConstantTensorOp> { using OpRewritePattern<ConstantTensorOp>::OpRewritePattern; public: LogicalResult matchAndRewrite(ConstantTensorOp cst_tensor_op, PatternRewriter &rewriter) const override { Location loc = cst_tensor_op.getLoc(); Type out_type = cst_tensor_op.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// The correct 'bias' broadcast will be synthesized manually. class ConvertBiasAddOp : public OpRewritePattern<TF::BiasAddOp> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::BiasAddOp op, PatternRewriter &rewriter) const override { Location loc = op.getLoc(); tensorflow::TensorFormat data_format;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/canonicalize.cc
// namespace mlir { namespace TFR { namespace { class UnrollSCFForOp : public OpRewritePattern<scf::ForOp> { using OpRewritePattern<scf::ForOp>::OpRewritePattern; public: LogicalResult matchAndRewrite(scf::ForOp for_op, PatternRewriter &rewriter) const override { Location loc = for_op.getLoc(); APInt lower_bound, upper_bound, step;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 14 22:15:06 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// converted to an ElementsAttr and then those are packed together using // tf.Pack op. struct ConvertConst : public OpConversionPattern<TF::ConstOp> { using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::ConstOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { // Verify that the tensor proto contains tensor of type variant and scalar
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/custom_call.h
namespace mlir { namespace odml { class ConvertCustomCallOp : public OpConversionPattern<mhlo::CustomCallOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( mhlo::CustomCallOp mhlo_custom_call, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final; }; // Ops that have a call_target_name starting with the prefix "custom_call." and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 02:41:24 UTC 2023 - 1.7K bytes - Viewed (0)