Class StorageConfig

tensorflow/core/framework/embedding/storage_config.h:22–44

namespace tensorflow {
namespace embedding {
struct StorageConfig {
  StorageConfig() : type(StorageType::DEFAULT),
                    path(""),
                    cache_strategy(CacheStrategy::LFU) {
    size = {1<<30,1<<30,1<<30,1<<30};
  }

  StorageConfig(StorageType t,
                const std::string& p,
                const std::vector<int64>& s,
                const EmbeddingConfig& ec,
                const CacheStrategy cache_strategy_ = CacheStrategy::LFU)
      : type(t),
        path(p),
        size(s),
        embedding_config(ec),
        cache_strategy(cache_strategy_) {}
  StorageType type;
  std::string path;
  std::vector<int64> size;
  CacheStrategy cache_strategy;
  EmbeddingConfig embedding_config;
};
} // namespace embedding
} // namespace tensorflow