Function RemoveKey

tensorflow/core/kernels/embedding_variable_ops_test.cc:1304–1309

}

void RemoveKey(EmbeddingVar<int64, float>* variable) {
  for (int64 i = 0; i < 10; i++) {
    sleep(1);
    variable->storage()->Remove(20);
  }
}

TEST(EmbeddingVariableTest, TestLookupRemoveConcurrency) {
  int value_size = 10;