for inference checkpoint (#30081)

* for inference checkpoint

Change-Id: I36c979240ffa55bf1ef0c9315402960762af6be4

* for inference checkpoint

Change-Id: I82025365d5b792cbea1ead506df685aecc8ac198
revert-31562-mean
tangwei12 4 years ago committed by GitHub
parent 7d4bdff07d
commit 7d1c149e09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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.", ""));
/* ========================================================================== */

Loading…
Cancel
Save