未验证 提交 7d1c149e 编写于 作者: T tangwei12 提交者: GitHub

for inference checkpoint (#30081)

* for inference checkpoint

Change-Id: I36c979240ffa55bf1ef0c9315402960762af6be4

* for inference checkpoint

Change-Id: I82025365d5b792cbea1ead506df685aecc8ac198
上级 7d4bdff0
......@@ -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.", ""));
/* ========================================================================== */
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册