- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for trackable_object_graph (0.26 sec)
-
tensorflow/cc/saved_model/bundle_v2.cc
RestoreObjectsCallback callback) { if (saved_object_graph().nodes_size() == 0 || trackable_object_graph().nodes_size() == 0) { return absl::OkStatus(); } // Start from root nodes of both the SavedObjectGraph and TrackableObjectGraph // and descend to leaves. Note that the TrackableObjectGraph can have cycles // (as can the SavedObjectGraph). // This is detected and cycle edges are skipped.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 18:28:37 UTC 2023 - 9.1K bytes - Viewed (0) -
tensorflow/cc/saved_model/bundle_v2.h
const SavedObjectGraph& saved_object_graph() { return meta_graph_def().object_graph_def(); } /// TrackableObjectGraph loaded from the variable_reader() checkpoint. TrackableObjectGraph& trackable_object_graph() { return trackable_object_graph_; } /// BundleReader for accessing the variables bundle. BundleReader* variable_reader() { return variable_reader_.get(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 20 03:32:59 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/cc/saved_model/bundle_v2_test.cc
SavedModelV2Bundle bundle; TF_ASSERT_OK(SavedModelV2Bundle::Load(export_dir, &bundle)); // Ensure that there are nodes in the trackable_object_graph. EXPECT_GT(bundle.trackable_object_graph().nodes_size(), 0); RestoreVarsAndVerify(&bundle, {"variable_x", "variable_y", "child_variable"}); } TEST_F(BundleV2Test, LoadsCyclicModule) { const std::string export_dir =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 12 17:51:15 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/tf_saved_model_api.cc
// The string key becomes the "name" attribute of the SerializedTensor proto // in the TrackableObjectGraph, // https://github.com/tensorflow/tensorflow/blob/ddc1bbad3dfd4a089eb96014f26cc16664b1b2f8/tensorflow/core/protobuf/trackable_object_graph.proto#L26 // And the checkpoint_key is a globally unique string derived from this name:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/cc/experimental/libexport/load.h
// Returns the path to the variables file. const std::string GetVariablesFilepath() const { return variables_filepath_; } private: SavedModel saved_model_proto_; TrackableObjectGraph trackable_object_graph_; std::unique_ptr<tensorflow::BundleReader> variable_reader_; std::string variables_filepath_; bool has_checkpoint_; absl::flat_hash_map<std::string, const NodeDef*> graph_def_nodes_by_name_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 13 06:33:42 UTC 2023 - 4.6K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.cc
#include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/protobuf/saved_object_graph.pb.h" #include "tensorflow/core/protobuf/struct.pb.h" #include "tensorflow/core/protobuf/trackable_object_graph.pb.h" namespace tf { namespace libtf { namespace runtime { using tensorflow::AbstractContext; using tensorflow::AbstractFunctionPtr; using tensorflow::DataType; using tensorflow::FunctionDef;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_model_utils.cc
#include "tensorflow/core/platform/stringpiece.h" #include "tensorflow/core/protobuf/saved_object_graph.pb.h" #include "tensorflow/core/protobuf/struct.pb.h" #include "tensorflow/core/protobuf/trackable_object_graph.pb.h" namespace tensorflow { namespace internal { namespace { using StructuredValueDictEntry = protobuf::MapPair<std::string, StructuredValue>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 19:17:46 UTC 2023 - 24K bytes - Viewed (0) -
tensorflow/cc/experimental/libexport/load.cc
return tf_package; } tensorflow::StatusOr<std::string> TFPackage::GetVariableCheckpointKey( int index) { // TODO(danielellis): make sure valid index const auto& trackable_object = trackable_object_graph_.nodes(index); const TrackableObjectGraph::TrackableObject::SerializedTensor* serialized_tensor = nullptr; for (auto& maybe_serialized_tensor : trackable_object.attributes()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 13 15:16:07 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} if (!value_it->second.has_tensor()) { return nullptr; } return &value_it->second.tensor(); } const TrackableObjectGraph::TrackableObject::SerializedTensor* FindSerializedTensorInTrackable( const TrackableObjectGraph::TrackableObject& trackable_object, StringPiece name) { for (const auto& maybe_serialized_tensor : trackable_object.attributes()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)