Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 29 for saver_def (0.16 sec)

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

      TF_ASSERT_OK_AND_ASSIGN(const std::optional<SaverDef> saver_def,
                              CreateSaverDef(control_ret_node_names, graph_def));
      ASSERT_NE(saver_def, std::nullopt);
      EXPECT_THAT(saver_def->version(), SaverDef::V2);
      EXPECT_THAT(saver_def->restore_op_name(), "restore_op_0");
      EXPECT_THAT(saver_def->filename_tensor_name(), "foo:0");
      EXPECT_THAT(saver_def->save_tensor_name(), "tf_quant__save_op_0:0");
    }
    
    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

      } else if (absl::c_none_of(fields, is_empty_predicate)) {
        SaverDef saver_def{};
        saver_def.set_version(SaverDef::V2);
        saver_def.set_filename_tensor_name(filename_tensor_name);
        saver_def.set_restore_op_name(restore_op_name);
        // :0 attached to indicate the first result tensor. This saves the model
        // checkpoint when fetched.
        saver_def.set_save_tensor_name(absl::StrCat(save_node_name, ":0"));
        return saver_def;
    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/python/save_model.py

              saved_model_constants.ASSETS_KEY,
              asset_any_proto,
          )
    
        model_saver = None
        # If `saver_def` is not None, it means there are variables in the graph.
        if saver_def:
          model_saver = saver.Saver(saver_def=saver_def)
          logging.info('Saver created with SaverDef: %s', saver_def)
    
          # Variables should be restored once before exporting as saved model
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/exported_model.proto

      // should be fed.
      repeated AssetFileDef asset_file_defs = 8;
    
      // SaverDef including the information required for saving and restoring
      // variables. This field is not set if there are no variables in the exported
      // graph. If set, the fields `version`, `filename_tensor_name`,
      // `restore_op_name` and `save_tensor_name` are populated.
      SaverDef saver_def = 10;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 06:12:59 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/python/py_function_lib.py

    ) -> Optional[saver_pb2.SaverDef]:
      """Returns the SaverDef from ExportedModel, None otherwise.
    
      Args:
        exported_model: ExportedModel to take the SaverDef from.
    
      Returns:
        SaverDef instance if the field `saver_def` is set. None otherwise.
      """
      if exported_model.HasField('saver_def'):
        return exported_model.saver_def
      return None
    
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 05:32:11 UTC 2024
    - 27.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h

        absl::string_view checkpoint_dir,
        std::optional<tensorflow::SaverDef> saver_def,
        const absl::flat_hash_map<std::string, std::string>& function_aliases,
        const std::vector<tensorflow::AssetFileDef>& asset_file_defs);
    
    // Creates a new `SaverDef` instance, which contains information regarding
    // checkpoint saving and restoring. This function returns a `SaverDef` instance
    // with four fields populated: `version`, `filename_tensor_name`,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  7. tensorflow/cc/saved_model/loader.cc

      if (meta_graph.has_saver_def()) {
        TF_RETURN_IF_ERROR(RunRestore(run_options, export_dir,
                                      meta_graph.saver_def().restore_op_name(),
                                      meta_graph.saver_def().filename_tensor_name(),
                                      asset_file_defs, session->get()));
      }
      // Record walltime spent in restoring graph from disk, but postpone metric
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 04:36:00 UTC 2024
    - 23K bytes
    - Viewed (0)
  8. tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/saved_model.pbtxt

        }
        node {
          name: "save/restore_all"
          op: "NoOp"
          input: "^save/restore_shard"
        }
        versions {
          producer: 23
        }
      }
      saver_def {
        filename_tensor_name: "save/Const:0"
        save_tensor_name: "save/Identity:0"
        restore_op_name: "save/restore_all"
        max_to_keep: 5
        sharded: true
        keep_checkpoint_every_n_hours: 10000.0
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 01:10:27 UTC 2017
    - 46.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        variable_and_assets.push_back(
            {input_.meta_graph_def().saver_def().filename_tensor_name(),
             variable_filename_op});
        variable_and_assets.insert(variable_and_assets.end(), assets.begin(),
                                   assets.end());
    
        const auto& restore_op_name =
            input_.meta_graph_def().saver_def().restore_op_name();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  10. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

    #include "tensorflow/core/platform/test.h"
    
    tensorflow::ServerDef GetServerDef(const std::string& job_name, int num_tasks) {
      tensorflow::ServerDef server_def;
      server_def.set_protocol("grpc");
      server_def.set_job_name(job_name);
      server_def.set_task_index(0);
      tensorflow::ClusterDef* cluster_def = server_def.mutable_cluster();
      tensorflow::JobDef* job_def = cluster_def->add_job();
      job_def->set_name(job_name);
      for (int i = 0; i < num_tasks; i++) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 22:09:57 UTC 2023
    - 6.7K bytes
    - Viewed (0)
Back to top