Searched refs:embedding_table_meta (Results 1 – 2 of 2) sorted by relevance
718 EmbeddingTableMeta embedding_table_meta; in HandleInitEmbeddings() local719 CHECK_RETURN_TYPE(embedding_table_meta.ParseFromArray(data.get(), SizeToInt(size))); in HandleInitEmbeddings()720 const Key &key = embedding_table_meta.key(); in HandleInitEmbeddings()726 embedding_table_meta.input_shape().begin(), embedding_table_meta.input_shape().end()); in HandleInitEmbeddings()729 embedding_table_meta.indices_shape().begin(), embedding_table_meta.indices_shape().end()); in HandleInitEmbeddings()732 embedding_table_meta.output_shape().begin(), embedding_table_meta.output_shape().end()); in HandleInitEmbeddings()738 const ParamInitInfoMessage &info = embedding_table_meta.info(); in HandleInitEmbeddings()
221 EmbeddingTableMeta embedding_table_meta; in InitPSEmbeddingTable() local222 embedding_table_meta.set_key(key); in InitPSEmbeddingTable()223 *embedding_table_meta.mutable_input_shape() = {input_shape.begin(), input_shape.end()}; in InitPSEmbeddingTable()224 *embedding_table_meta.mutable_indices_shape() = {indices_shape.begin(), indices_shape.end()}; in InitPSEmbeddingTable()225 *embedding_table_meta.mutable_output_shape() = {output_shape.begin(), output_shape.end()}; in InitPSEmbeddingTable()226 *embedding_table_meta.mutable_info() = info; in InitPSEmbeddingTable()228 std::string kv_data = embedding_table_meta.SerializeAsString(); in InitPSEmbeddingTable()