- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for MatchAnyOpTypeTag (0.22 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
} void runOnOperation() override; }; class RemoveTpuOp : public RewritePattern { public: explicit RemoveTpuOp(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} private: LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override {
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/tf2xla/transforms/legalize_tf_with_tf2xla.cc
const std::string& device_type, bool prefer_tf2xla) : ConversionPattern(converter, MatchAnyOpTypeTag(), /*benefit=*/1, ctx), device_type_(device_type), prefer_tf2xla_(prefer_tf2xla) {} LogicalResult matchAndRewrite( Operation* op, ArrayRef<Value> operands,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/cast_bf16_ops_to_f32.cc
} void runOnOperation() override; }; class CastBf16OpsToF32 : public RewritePattern { public: explicit CastBf16OpsToF32(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} private: LogicalResult match(Operation* op) const override { if (isa<TF::CastOp, TF::ConstOp>(op) || op->getName().hasTrait<OpTrait::ZeroOperands>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
} } return std::nullopt; } class AddQuantizationUnitLoc : public RewritePattern { public: explicit AddQuantizationUnitLoc(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} private: LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override { if (!IsOpWithQuantizableTrait(op) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
namespace mlir { namespace { class ConvertResultsBroadcastableShapeOp : public RewritePattern { public: ConvertResultsBroadcastableShapeOp(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), 1, context) {} LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override; private: template <typename Op>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
BFloat16TypeConverter& converter_; }; class BFloat16TypePattern : public ConversionPattern { public: BFloat16TypePattern(TypeConverter& converter, MLIRContext* ctx) : ConversionPattern(converter, MatchAnyOpTypeTag(), /*benefit=*/1, ctx) {} LogicalResult matchAndRewrite( Operation* op, const ArrayRef<Value> operands, ConversionPatternRewriter& rewriter) const override { if (getTypeConverter()->isLegal(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
// identifying `erf` ops and validate the structure around them. class OutlineGELU : public RewritePattern { public: explicit OutlineGELU(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override { if (!MatchERF(op)) return failure(); // `add 1`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0)