Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for serialize_stablehlo_ops (0.96 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_export.h

                                           const FlatbufferExportOptions& options,
                                           std::string* serialized_flatbuffer,
                                           bool serialize_stablehlo_ops = false);
    }  // namespace tflite
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 09 02:51:43 UTC 2023
    - 2.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.h

    extern llvm::cl::opt<bool> post_training_quantization;
    
    // TF to stablehlo pass flags
    extern llvm::cl::opt<bool> enable_stablehlo_conversion;
    
    // Wether serialize stablehlo ops or not
    extern llvm::cl::opt<bool> serialize_stablehlo_ops;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h

        const mlir::TFL::PassConfig& pass_config,
        const std::unordered_set<std::string>& saved_model_tags,
        llvm::StringRef saved_model_dir, SavedModelBundle* saved_model_bundle,
        std::string* result, bool serialize_stablehlo_ops = false,
        const quantization::PyFunctionLibrary* quantization_py_function_lib =
            nullptr);
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 08:30:24 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

        const mlir::TFL::PassConfig& pass_config,
        const std::unordered_set<std::string>& saved_model_tags,
        llvm::StringRef saved_model_dir, SavedModelBundle* saved_model_bundle,
        std::string* result, bool serialize_stablehlo_ops,
        const PyFunctionLibrary* quantization_py_function_lib) {
      // Explicitly disable dumping Op details on failures.
      module.getContext()->printOpOnDiagnostic(false);
    
      mlir::DialectRegistry registry;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc

        llvm::cl::desc("Convert \"tf.TensorList*\" ops to \"tfl.custom_op\""
                       "if they can all be supported."),
        llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<bool> serialize_stablehlo_ops(
        "serialize-stablehlo-ops",
        llvm::cl::desc("Wether serialize stablehlo ops or not"),
        llvm::cl::init(true));
    
    // NOLINTNEXTLINE
    opt<bool> reduce_type_precision(
        "reduce-type-precision",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

      auto status = tensorflow::ConvertTFExecutorToTFLOrFlatbuffer(
          module.value().get(), output_mlir, toco_flags, pass_config, tags,
          /*saved_model_dir=*/"", bundle.get(), &result, serialize_stablehlo_ops);
      if (!status.ok()) {
        llvm::errs() << status.message() << '\n';
        return kTrFailure;
      }
    
      std::string error_msg;
      auto output = mlir::openOutputFile(output_file_name, &error_msg);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc

      auto status = ConvertTFExecutorToTFLOrFlatbuffer(
          module.get(), /*export_to_mlir=*/false, toco_flags, pass_config_copy,
          saved_model_tags, model_flags.saved_model_dir(), saved_model_bundle,
          result, /*serialize_stablehlo_ops=*/false, quantization_py_function_lib);
      if (toco_flags.has_dump_graphviz_dir()) {
        TF_RETURN_IF_ERROR(DumpOpGraphToFile(
            // rename once we enable the new converter feature flag.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 17.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

                                           std::string* serialized_flatbuffer,
                                           bool serialize_stablehlo_ops) {
      auto maybe_translated = Translator::Translate(
          module, options.toco_flags, options.saved_model_tags,
          options.op_or_arg_name_mapper, options.metadata, serialize_stablehlo_ops,
          options.custom_option_alignment);
      if (!maybe_translated) return false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top