Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for restored_resources (0.21 sec)

  1. tensorflow/c/experimental/saved_model/core/revived_types/partially_revived_objects.cc

          }
          *handle = assets_iter->second->handle();
          return Status();
        }
        case SavedObject::kResource: {
          const auto& resource_iter = objects.restored_resources.find(node_id);
          if (resource_iter == objects.restored_resources.end()) {
            return absl::FailedPreconditionError(absl::StrCat(
                "Tried to convert node id ", node_id,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 20:11:48 UTC 2023
    - 23.7K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/saved_model/core/revived_types/revived_objects.h

      gtl::FlatMap<int, std::unique_ptr<TFSignatureDefFunction>>
          signature_def_functions;
      RevivedObjectContainer<TFConcreteFunction> concrete_functions;
      gtl::FlatMap<int, RestoredResource> restored_resources;
      gtl::FlatMap<std::string, int> signatures_map;
    };
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 05 23:10:04 UTC 2020
    - 4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/saved_model/core/revived_types/partially_revived_objects.h

      gtl::FlatMap<int, TFConcreteFunctionRevivalState> concrete_functions;
      gtl::FlatMap<int, TFSignatureDefFunctionRevivalState> signature_def_functions;
      gtl::FlatMap<int, RestoredResourceRevivalState> restored_resources;
      gtl::FlatMap<std::string, int> signatures_map;
    
      Status Build(ImmediateExecutionContext* ctx,
                   const SavedObjectGraph& obj_graph, RevivedObjects* revived);
    };
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 30 23:21:36 UTC 2020
    - 3.3K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/saved_model/core/saved_model_utils.cc

          // We'll set the resource's functions in a subsequent pass, once we get
          // all functions in a partially revived state.
          resource_revival_state.device = node.resource().device();
          objects->restored_resources[i] = std::move(resource_revival_state);
        } else if (node.kind_case() == SavedObject::kFunction) {
          // Get the SavedFunction node and skip if it has no concrete functions.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 12 19:17:46 UTC 2023
    - 24K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/saved_model/core/tf_saved_model_api.cc

          }));
    
      return Status();
    }
    
    Status InitializeAllResources(const RevivedObjects& revived) {
      for (const auto& node_and_resource : revived.restored_resources) {
        const RestoredResource& resource = node_and_resource.second;
        TF_RETURN_IF_ERROR(resource.Initialize());
      }
      return Status();
    }
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 09:34:33 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/saved_model/core/revived_types/restored_resource.cc

          device_(device),
          create_resource_(create_resource),
          initialize_(initialize),
          destroy_resource_(destroy_resource) {}
    
    Status RestoredResource::Initialize() const {
      return ExecuteNoArgDummyReturnFunction(initialize_);
    }
    
    RestoredResource::~RestoredResource() {
      // Note(bmzhao): SavedModels saved before
      // https://github.com/tensorflow/tensorflow/commit/3c806101f57768e479f8646e7518bbdff1632ca3
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/saved_model/core/revived_types/restored_resource.h

      RestoredResource& operator=(RestoredResource&& other) = default;
    
      ~RestoredResource() override;
    
     private:
      std::string device_;
      TFConcreteFunction* create_resource_;
      TFConcreteFunction* initialize_;
      TFConcreteFunction* destroy_resource_;
    };
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 23 04:49:47 UTC 2020
    - 4.6K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/saved_model/core/revived_types/BUILD

            "@com_google_absl//absl/strings",
            "@com_google_absl//absl/types:span",
        ],
    )
    
    cc_library(
        name = "restored_resource",
        srcs = [
            "restored_resource.cc",
        ],
        hdrs = [
            "restored_resource.h",
        ],
        deps = [
            ":tensorhandle_convertible",
            ":tf_concrete_function",
            "//tensorflow/c/eager:abstract_tensor_handle",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 11 05:09:48 UTC 2023
    - 7.9K bytes
    - Viewed (0)
Back to top