- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetHardwareRewritePatterns (0.66 sec)
-
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) -
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) -
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)