Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for init_node_name (0.22 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc

          std::move(graph_def), "init_node_name", "checkpoint_dir",
          /*saver_def=*/std::nullopt,
          /*function_aliases=*/{}, /*asset_file_defs=*/{});
      ASSERT_THAT(exported_model.graph_def().node(), SizeIs(1));
      EXPECT_THAT(exported_model.graph_def().node()[0].name(), StrEq("foo"));
    
      EXPECT_THAT(exported_model.init_node_name(), StrEq("init_node_name"));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 19.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc

      }
      const std::string init_node_name =
          GetNodeName(control_ret_node_names, kTfSavedModelInitializerInitType);
    
      TF_ASSIGN_OR_RETURN(const std::optional<SaverDef> saver_def,
                          CreateSaverDef(control_ret_node_names, graph_def));
    
      return CreateExportedModelFromGraphDef(std::move(graph_def), init_node_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/exported_model.proto

      reserved 'file_prefix_tensor_name';
    
      GraphDef graph_def = 1;
    
      // Name of the initialization node (TF Operation) used for initializing
      // resources like hash tables upon loading.
      string init_node_name = 2;
    
      // Path to the directory where checkpoint files are saved. This directoy is
      // not expected to be persistent (usually a temporary directory). When
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 06:12:59 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h

    // Factory function for `ExportedModel`.
    [[nodiscard]] tensorflow::quantization::ExportedModel
    CreateExportedModelFromGraphDef(
        tensorflow::GraphDef&& graph_def, absl::string_view init_node_name,
        absl::string_view checkpoint_dir,
        std::optional<tensorflow::SaverDef> saver_def,
        const absl::flat_hash_map<std::string, std::string>& function_aliases,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/python/py_function_lib.py

      """
      save_model.save_model_v1(
          exported_model.graph_def,
          dst_saved_model_path,
          signature_def_map,
          tags,
          init_op_name=exported_model.init_node_name,
          saver_def=_get_saver_def_or_none(exported_model),
          checkpoint_dir=exported_model.checkpoint_dir,
          function_aliases=exported_model.function_aliases,
          asset_file_defs=exported_model.asset_file_defs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 05:32:11 UTC 2024
    - 27.4K bytes
    - Viewed (0)
Back to top