Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for export_original_tf_func_name (0.52 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc

      if (!module) return failure();
    
      tensorflow::GraphExportConfig confs;
      confs.export_entry_func_to_flib = export_entry_func_to_flib;
      confs.export_original_tf_func_name = export_original_tf_func_name;
    
      std::unique_ptr<tensorflow::FunctionLibraryDefinition> flib_def;
      auto graph =
          std::make_unique<tensorflow::Graph>(tensorflow::OpRegistry::Global());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.h

    extern llvm::cl::opt<bool> enable_soft_placement;
    extern llvm::cl::opt<bool> set_original_tf_func_name;
    
    // Export options.
    extern llvm::cl::opt<bool> export_entry_func_to_flib;
    extern llvm::cl::opt<bool> export_original_tf_func_name;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 10 20:59:50 UTC 2023
    - 2.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc

        "tf-export-entry-func-to-flib",
        llvm::cl::desc(
            "Export entry function to function library instead of graph"),
        llvm::cl::init(false));
    // NOLINTNEXTLINE
    opt<bool> export_original_tf_func_name(
        "tf-export-original-func-name",
        llvm::cl::desc("Export functions using the name set in the attribute "
                       "'tf._original_func_name' if it exists."),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 10 20:59:50 UTC 2023
    - 5.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h

      // graph.
      bool export_entry_func_to_flib = false;
      // Whether to export functions using the name set in the attribute
      // `tf._original_func_name` if it exists.
      bool export_original_tf_func_name = false;
    };
    
    // Parses the command line flag strings to the specification of nodes in
    // the Graph.
    Status ParseOutputArrayInfo(absl::string_view array_names,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 04:56:10 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

    };
    
    std::string FindFunctionName(const GraphExportConfig& configs, FuncOp func) {
      if (auto original_func_name =
              func->getAttrOfType<mlir::StringAttr>("tf._original_func_name");
          configs.export_original_tf_func_name && original_func_name) {
        return original_func_name.str();
      }
    
      return func.getName().str();
    }
    
    absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetArgumentNode(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

    };
    
    std::string FindFunctionName(const GraphExportConfig& configs, FuncOp func) {
      if (auto original_func_name =
              func->getAttrOfType<mlir::StringAttr>("tf._original_func_name");
          configs.export_original_tf_func_name && original_func_name) {
        return original_func_name.str();
      }
    
      return func.getName().str();
    }
    
    absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetArgumentNode(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
Back to top