Searched refs:table_lookup_op (Results 1 – 1 of 1) sorted by relevance
383 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in DoEmbeddingLookup() local384 MS_EXCEPTION_IF_NULL(table_lookup_op); in DoEmbeddingLookup()391 table_lookup_op->ReInit(shapes); in DoEmbeddingLookup()393 const std::vector<size_t> output_shapes = table_lookup_op->output_sizes(); in DoEmbeddingLookup()423 table_lookup_op->Execute(inputs, workspaces, outputs); in DoEmbeddingLookup()439 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in UpdateEmbeddings() local440 MS_EXCEPTION_IF_NULL(table_lookup_op); in UpdateEmbeddings()441 …table_lookup_op->UpdateEmbeddings(table_ptr->data(), lookup_ids.data(), vals.data(), lookup_ids.si… in UpdateEmbeddings()