Searched refs:new_tensor_size (Results 1 – 1 of 1) sorted by relevance
558 size_t new_tensor_size = static_cast<size_t>(new_tensor->data().nbytes()); in SyncEmbeddingTables() local560 if (new_tensor_size != embedding_table_size) { in SyncEmbeddingTables()561 … MS_LOG(EXCEPTION) << "Shape of embedding table can't match. New tensor size:" << new_tensor_size in SyncEmbeddingTables()567 CopyTensorData(new_tensor_data_ptr, new_tensor_size, weights_[key]->data()); in SyncEmbeddingTables()