- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 37 for populateWithGenerated (0.19 sec)
-
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
custom_op_map_); } RewritePatternSet patterns(&getContext()); auto func = getOperation(); auto* ctx = func.getContext(); TFL::populateWithGenerated(patterns); patterns.add<quant::FoldTrivalRequantizeOp<QuantizeOp>>(ctx); patterns.add<PruneUnusedOpsWithSideEffect<TFL::LSTMOp>>(ctx); patterns.add<PruneUnusedOpsWithSideEffect<TFL::UnidirectionalSequenceLSTMOp>>(
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/tensorflow/passes/cast_bf16_ops_to_f32.cc
void CastBf16OpsToF32Pass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx); auto module_op = getOperation(); patterns.add<CastBf16OpsToF32>(ctx); populateWithGenerated(patterns); if (failed(applyPatternsAndFoldGreedily(module_op, std::move(patterns)))) { module_op.emitError() << "quant-cast-bf16-ops-to-f32 failed."; signalPassFailure(); } }
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/lite/stablehlo/transforms/tflite_legalize_hlo.cc
} } // namespace void PopulateLegalizeHloToTFLitePatterns(RewritePatternSet* patterns, MLIRContext* context) { patterns->add<odml::ConvertCustomCallOp>(context); populateWithGenerated(*patterns); patterns->add<ConvertReduceOpToTFLiteArgmin, ConvertReduceOpToTFLiteArgmax>( context); } // Creates an instance of the pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/post_quantize.cc
auto func = getOperation(); auto* ctx = func.getContext(); patterns.add<FoldTrivalRequantizeOp<quantfork::QuantizeCastOp>, RemoveVolatileOps<kPreserveNone>, RemoveRedundantScast>(ctx); populateWithGenerated(patterns); if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) { signalPassFailure(); } } } // namespace
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/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc
void LiftQuantizableSpotsAsFunctionsDRQPass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx); ModuleOp module = getOperation(); populateWithGenerated(patterns); patterns.add<CheckQuantizableOps>(ctx, quantization_method_, target_opset_, min_num_elements_for_weights_); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
struct TensorFlowOptimizePass : public impl::TensorFlowOptimizePassBase<TensorFlowOptimizePass> { LogicalResult initialize(MLIRContext *context) override { RewritePatternSet pattern_list(context); populateWithGenerated(pattern_list); pattern_list.add<SimplifyBroadcastReshape>(context); patterns = std::move(pattern_list); return success(); } void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.cc
} // namespace void PopulateDecomposeResourceOpsPatterns(MLIRContext *context, RewritePatternSet *patterns) { patterns->add<DecomposeRngReadAndSkipOp>(context); populateWithGenerated(*patterns); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
auto* ctx = &getContext(); RewritePatternSet patterns(ctx); patterns.add<ConvertBatchMatMulOp2FullyConnectedOp, ConvertBatchMatMulOpToReduceSum>(ctx); TFL::populateWithGenerated(patterns); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateOptimizeBatchMatmulPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
void PrepareQuantizeDRQPass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx); ModuleOp module_op = getOperation(); populateWithGenerated(patterns); patterns.add<PrepareDRQQuantizableOp>(ctx, quant_specs_, op_set_, enable_per_channel_quantization_); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize.cc
} const quant::QuantPassSpec quant_params = { {quant_specs.verify_numeric, error_tolerance_, quant_specs.whole_model_verify, enable_log_if_failed_}, quant_specs}; populateWithGenerated(patterns); if (quant_specs.weight_quantization || quant_specs.use_fake_quant_num_bits || quant_specs.qdq_conversion_mode == quant::QDQConversionMode::kQDQDynamic) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 13.3K bytes - Viewed (0)