Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for populateCanonicalizationPatterns (0.26 sec)

  1. tensorflow/compiler/mlir/tfr/passes/passes.h

    namespace mlir {
    namespace TFR {
    
    // Scans the func op and adds all the canonicalization patterns of the ops
    // except the tf ops, inside the function.
    void populateCanonicalizationPatterns(func::FuncOp func,
                                          RewritePatternSet &patterns);
    
    // Decompose ops.
    std::unique_ptr<OperationPass<func::FuncOp>> CreateDecomposeTFOpsPass(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/passes/canonicalize.cc

        return failure();
      }
    
      // If the inlining was successful then erase the scf.if op.
      rewriter.eraseOp(inline_point);
      return success();
    }
    
    }  // namespace
    
    void populateCanonicalizationPatterns(func::FuncOp func,
                                          RewritePatternSet &patterns) {
      MLIRContext *context = func.getContext();
      mlir::Dialect *tf = context->getLoadedDialect<mlir::TF::TensorFlowDialect>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 14 22:15:06 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/passes/decompose.cc

    void DecomposeTFOpsPass::ApplyCanonicalization() {
      func::FuncOp func = getOperation();
      RewritePatternSet patterns(&getContext());
    
      populateWithGenerated(patterns);
      populateCanonicalizationPatterns(func, patterns);
    
      (void)applyPatternsAndFoldGreedily(func, std::move(patterns));
    }
    
    LogicalResult DecomposeTFOpsPass::RewriteUnregisteredTFOps() {
      func::FuncOp func = getOperation();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

                            : func->getParentOfType<ModuleOp>());
    
      RewritePatternSet patterns(&getContext());
      patterns.add<RewriteTFRCallOp>(ctx, table, materialize_derived_attrs_);
    
      populateCanonicalizationPatterns(func, patterns);
    
      (void)applyPatternsAndFoldGreedily(func, std::move(patterns));
    }
    }  // namespace
    
    // Creates an instance of the pass to raise TFR call ops to the TF ops.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
Back to top