Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetHardwareRewritePatterns (0.35 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.h

    namespace tac {
    
    // Returns true if 'op' is supported to run on 'hardware'.
    bool IsSupported(Operation* op, const std::string& hardware);
    
    // Return proper rewriter patterns for different hardwares.
    RewritePatternSet GetHardwareRewritePatterns(MLIRContext* context,
                                                 const std::string& hardware);
    
    // Convert quantized ops to float, this will essentially insert dequantize &
    // quantize pair around the op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 07 18:43:51 UTC 2022
    - 2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

    namespace mlir {
    namespace TFL {
    namespace tac {
    namespace {
    #include "tensorflow/compiler/mlir/lite/experimental/tac/transforms/generated_transform_patterns.inc"
    }  // namespace
    
    RewritePatternSet GetHardwareRewritePatterns(MLIRContext* context,
                                                 const std::string& hardware) {
      auto* devce_hardware = GetTargetHardware(hardware);
      if (devce_hardware == nullptr) return {context};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/transforms/get_alternative_subgraph.cc

    }
    
    void AlternativeSubgraphPass::Optimize(func::FuncOp func,
                                           const std::string& hardware) {
      auto* ctx = &getContext();
      RewritePatternSet patterns = GetHardwareRewritePatterns(ctx, hardware);
      (void)applyPatternsAndFoldGreedily(func, std::move(patterns));
    }
    
    // Get the alternative view of the func for the given device_inference_type.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 12.3K bytes
    - Viewed (0)
Back to top