Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for init_node_name (0.2 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/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)
  4. 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