|
|
|
@ -17,6 +17,7 @@ limitations under the License. */
|
|
|
|
|
#include <memory>
|
|
|
|
|
|
|
|
|
|
#include "paddle/fluid/framework/no_need_buffer_vars_inference.h"
|
|
|
|
|
#include "paddle/fluid/framework/op_version_registry.h"
|
|
|
|
|
#include "paddle/fluid/framework/var_type_inference.h"
|
|
|
|
|
|
|
|
|
|
namespace paddle {
|
|
|
|
@ -224,3 +225,16 @@ REGISTER_OP_CPU_KERNEL(lookup_table, ops::LookupTableKernel<float>,
|
|
|
|
|
ops::LookupTableKernel<int8_t>);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(lookup_table_grad, ops::LookupTableGradKernel<float>,
|
|
|
|
|
ops::LookupTableGradKernel<double>);
|
|
|
|
|
|
|
|
|
|
/* ========================== register checkpoint ===========================*/
|
|
|
|
|
|
|
|
|
|
REGISTER_OP_VERSION(lookup_table)
|
|
|
|
|
.AddCheckpoint(
|
|
|
|
|
R"ROC(
|
|
|
|
|
Upgrade lookup_table add 1 attribute [entry_config].
|
|
|
|
|
)ROC",
|
|
|
|
|
paddle::framework::compatible::OpVersionDesc().NewAttr(
|
|
|
|
|
"entry_config",
|
|
|
|
|
"(std::string) embedding sparse feature entry config.", ""));
|
|
|
|
|
|
|
|
|
|
/* ========================================================================== */
|
|
|
|
|