revert-3824-remove_grad_op_type
fengjiayi 8 years ago
parent 19dfe1f38b
commit fb6bec6a8f

@ -222,6 +222,9 @@ class OpRegistry {
varmap[var.name()] = idx++;
}
// ================================================ //
} else {
op_info.proto_ = nullptr;
op_info.checker_ = nullptr;
}
op_info_map().insert(std::make_pair(op_type, op_info));
// register gradient op

@ -176,6 +176,9 @@ All parameter, weight, gradient are variables in Paddle.
std::vector<py::bytes> ret_values;
for (auto it = op_info_map.begin(); it != op_info_map.end(); ++it) {
const OpProto *proto = it->second.proto_;
if (proto == nullptr) {
continue;
}
PADDLE_ENFORCE(proto->IsInitialized(), "OpProto must all be initialized");
std::string str;
PADDLE_ENFORCE(proto->SerializeToString(&str),

Loading…
Cancel
Save