Function CreateEmbeddingVar

tensorflow/core/kernels/embedding_variable_test.h:106–148

}

EmbeddingVar<int64, float>* CreateEmbeddingVar(
    int value_size, Tensor& default_value,
    int64 default_value_dim, int64 filter_freq = 0,
    int64 steps_to_live = 0,
    float l2_weight_threshold=-1.0,
    embedding::StorageType storage_type = embedding::StorageType::DRAM,
    std::vector<int64> storage_size = {1024*1024*1024,
                                       1024*1024*1024,
                                       1024*1024*1024,
                                       1024*1024*1024},
    bool record_freq = false,
    int64 max_element_size = 0,
    float false_positive_probability = -1.0,
    DataType counter_type = DT_UINT64) {
  auto embedding_config = EmbeddingConfig(
      0, 0, 1, 0, "emb_var", steps_to_live,
      filter_freq, 999999, l2_weight_threshold,
      max_element_size, false_positive_probability,
      counter_type, default_value_dim,
      0.0, record_freq, false, false);
  auto feat_desc = new embedding::FeatureDescriptor<float>(
      1, 1, ev_allocator(), storage_type,
      record_freq,
      embedding_config.is_save_version(),
      {embedding_config.is_counter_filter(), filter_freq});
  auto storage =
      embedding::StorageFactory::Create<int64, float>(
          embedding::StorageConfig(
              storage_type, "",
              storage_size,
              embedding_config),
          cpu_allocator(),
          feat_desc,
          "emb_var");
	auto ev = new EmbeddingVar<int64, float>(
      "emb_var",
      storage,
      embedding_config,
      cpu_allocator(),
      feat_desc);
	ev->Init(default_value, default_value_dim);
  return ev;
}
} //namespace embedding
} //namespace tensorflow
#endif // TENSORFLOW_CORE_FRAMEWORK_EMBEDDING_STORAGE_FACTORY_H_