- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetAcceptedStableHLODialects (0.33 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/check_accepted_ops_pass.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CheckAcceptedOpsPass) explicit CheckAcceptedOpsPass( const std::vector<std::string> &optional_accepted_dialects) : accepted_dialects_(GetAcceptedStableHLODialects()), optional_accepted_dialects_(optional_accepted_dialects) {} // Check if TF dialect ops exist over the module. void runOnOperation() override; private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 21:06:11 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/stablehlo_util.h
#include <string> #include <vector> #include "llvm/ADT/StringRef.h" namespace mlir { namespace odml { std::vector<std::string> GetAcceptedStableHLODialects(); std::vector<std::string> GetAcceptedTFLiteDialects(); // Can we find the given `dialect_name` in the `accepted_dialects`? bool IsAcceptedDialect(llvm::StringRef dialect_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 21:06:11 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/stablehlo_util.cc
#include <algorithm> #include <string> #include <vector> #include "llvm/ADT/DenseSet.h" namespace mlir { namespace odml { std::vector<std::string> GetAcceptedStableHLODialects() { // It returns the default list of accepted dialects. std::vector<std::string> accepted_dialects({"stablehlo", "builtin", "func"}); return accepted_dialects; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 21:06:11 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
// Because this pass aborts the pass if there are unconverted ops, // we need to locate createPrintOpStatsPass after all optimization. pm.addPass( mlir::odml::createPrintOpStatsPass(GetAcceptedStableHLODialects())); } if (failed(pm.run(tf_module))) { return tensorflow::errors::Aborted("Lowering to StableHLO failed."); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
/*smuggle_disallowed_ops=*/true); // Print out a detailed report of non-converted stats. pass_manager.addPass(mlir::odml::createPrintOpStatsPass( mlir::odml::GetAcceptedStableHLODialects())); mlir::odml::AddStablehloOptimizationPasses(pass_manager); if (toco_flags.has_quantization_options()) { stablehlo::quantization::AddQuantizationPasses(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0)