|
|
|
@ -43,13 +43,11 @@ class SingleOpInplaceInToOut : public InplaceOpInference {
|
|
|
|
|
public:
|
|
|
|
|
std::unordered_map<std::string, std::string> operator()(
|
|
|
|
|
const OpDesc& op_desc, bool use_cuda) const override {
|
|
|
|
|
PADDLE_ENFORCE_EQ(op_desc.InputNames().size(), 1,
|
|
|
|
|
"Op inputs must be unique");
|
|
|
|
|
PADDLE_ENFORCE_EQ(op_desc.OutputNames().size(), 1,
|
|
|
|
|
"Op outputs must be unique");
|
|
|
|
|
auto x_name = op_desc.InputNames().at(0);
|
|
|
|
|
auto out_name = op_desc.OutputNames().at(0);
|
|
|
|
|
return std::unordered_map<std::string, std::string>{{x_name, out_name}};
|
|
|
|
|
auto inputs = op_desc.InputNames();
|
|
|
|
|
auto outputs = op_desc.OutputNames();
|
|
|
|
|
PADDLE_ENFORCE_EQ(inputs.size(), 1, "Op inputs must be unique");
|
|
|
|
|
PADDLE_ENFORCE_EQ(outputs.size(), 1, "Op outputs must be unique");
|
|
|
|
|
return {{inputs[0], outputs[0]}};
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|