Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ExportModule (0.15 sec)

  1. tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo_translate.cc

        llvm::cl::Optional, llvm::cl::init(false));
    
    }  // namespace
    
    namespace mlir {
    
    namespace {
    // Dump the ModuleOp 'module' to the file specified using 'outputFileName'
    absl::Status ExportModule(ModuleOp module) {
      std::string error_msg;
      auto output = openOutputFile(output_filename, &error_msg);
      if (output == nullptr) {
        return absl::AbortedError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 22:58:42 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo_lib.cc

      if (mlir::failed(mlir::writeBytecodeToFile(module, os, config))) {
        return absl::InvalidArgumentError("mlir::writeBytecodeToFile failed");
      }
      return bytecode;
    }
    
    absl::StatusOr<std::string> ExportModule(ModuleOp module) {
      const std::string output_filename = tensorflow::io::GetTempFilename(".mlir");
      std::string error_msg;
      auto output = openOutputFile(output_filename, &error_msg);
      if (output == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 22:58:42 UTC 2024
    - 5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc

      auto output_export_status =
          ExportModule(*module, output_path, elide_large_elements_attrs);
      if (!conversion_status.ok()) {
        LOG(ERROR) << "TF to StableHLO conversion failed: "
                   << conversion_status.message();
    
        auto debug_export_status = ExportModule(
            *module, absl::StrCat(verbose_dir, "/debug_stablehlo.mlir"),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 14.1K bytes
    - Viewed (0)
Back to top