Searched defs:update_vals (Results 1 – 3 of 3) sorted by relevance
41 …l void UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, const float *update_vals, in UpdateEmbeddings()
91 const float *update_vals, size_t ids_size) { in UpdateEmbeddings()
818 const Values &update_vals = {input.values().begin(), input.values().end()}; in HandleUpdateEmbeddings() local