|
|
|
@ -81,6 +81,8 @@ class LookupTableOpMaker : public framework::OpProtoAndCheckerMaker {
|
|
|
|
|
"Otherwise the given value indicates padding the output "
|
|
|
|
|
"with zeros whenever lookup encounters it in Ids.")
|
|
|
|
|
.SetDefault(kNoPadding);
|
|
|
|
|
// NOTE(minqiyang): grad_inplace is an temporal attribute,
|
|
|
|
|
// please do NOT set this attribute in python layer.
|
|
|
|
|
AddAttr<bool>("grad_inplace",
|
|
|
|
|
"(boolean, default false) "
|
|
|
|
|
"If the grad op reuse the input's variable.")
|
|
|
|
|