Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for hlo_import_type (0.13 sec)

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

    extern llvm::cl::opt<std::string> saved_model_tags;
    extern llvm::cl::opt<std::string> saved_model_exported_names;
    
    // Import HLO.
    enum HloImportType { proto, hlotxt, mlir_text };
    
    extern llvm::cl::opt<bool> import_hlo;
    extern llvm::cl::opt<HloImportType> hlo_import_type;
    
    // enable_hlo_to_tf_conversion and disable_hlo_to_tfl_conversion are used to
    // control the HLO to TF and HLO to TFLite conversion while debugging an
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

                       << error;
          return kTrFailure;
        }
    
        auto content = buffer->getBuffer();
        if (hlo_import_type == HloImportType::hlotxt) {
          module = xla::HloTextToMlirHloTranslateFunction(content, &context, false);
        } else if (hlo_import_type == HloImportType::proto) {
          module = xla::HloToMlirHloTranslateFunction(content, &context, false);
        } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc

    // NOLINTNEXTLINE
    opt<bool> import_hlo("import-hlo",
                         llvm::cl::desc("Whether the input file is hlo file."),
                         llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<HloImportType> hlo_import_type(
        "hlo-import-type", llvm::cl::desc("The file type of the hlo."),
        llvm::cl::values(clEnumVal(proto, "Import hlo in proto binary format"),
                         clEnumVal(hlotxt, "Import hlo in hlotxt format"),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 7.9K bytes
    - Viewed (0)
Back to top