Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for trackable_object_graph (0.41 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top