提交 914a2f2b 编写于 作者: F fengjiayi

Format code

上级 39c986c7
...@@ -59,10 +59,9 @@ REGISTER_OP(io_ignored, f::NOP, f::IOIgnoredOpMaker, io_ignored_grad, f::NOP); ...@@ -59,10 +59,9 @@ REGISTER_OP(io_ignored, f::NOP, f::IOIgnoredOpMaker, io_ignored_grad, f::NOP);
TEST(GradOpBuilder, MutiInOut) { TEST(GradOpBuilder, MutiInOut) {
std::shared_ptr<f::OperatorBase> test_op(f::OpRegistry::CreateOp( std::shared_ptr<f::OperatorBase> test_op(f::OpRegistry::CreateOp(
"mult_io", "mult_io", {{"In1", {"in1"}},
{{"In1", {"in1"}}, {"In2_mult", {"in2_1", "in2_2", "in2_3"}},
{"In2_mult", {"in2_1", "in2_2", "in2_3"}}, {"In3", {"in3"}}},
{"In3", {"in3"}}},
{{"Out1", {"out1"}}, {"Out2_mult", {"out2_1", "out2_2"}}}, {})); {{"Out1", {"out1"}}, {"Out2_mult", {"out2_1", "out2_2"}}}, {}));
std::shared_ptr<f::OperatorBase> grad_test_op = std::shared_ptr<f::OperatorBase> grad_test_op =
f::OpRegistry::CreateGradOp(*test_op); f::OpRegistry::CreateGradOp(*test_op);
...@@ -92,10 +91,9 @@ TEST(GradOpBuilder, MutiInOut) { ...@@ -92,10 +91,9 @@ TEST(GradOpBuilder, MutiInOut) {
TEST(GradOpBuilder, IOIgnoredInGradient) { TEST(GradOpBuilder, IOIgnoredInGradient) {
std::shared_ptr<f::OperatorBase> test_op(f::OpRegistry::CreateOp( std::shared_ptr<f::OperatorBase> test_op(f::OpRegistry::CreateOp(
"io_ignored", "io_ignored", {{"In1", {"in1"}},
{{"In1", {"in1"}}, {"In2_mult", {"in2_1", "in2_2"}},
{"In2_mult", {"in2_1", "in2_2"}}, {"In3_mult", {"in3_1", "in3_2"}}},
{"In3_mult", {"in3_1", "in3_2"}}},
{{"Out1_mult", {"out1_1", "out1_2"}}, {"Out2", {"out2"}}}, {})); {{"Out1_mult", {"out1_1", "out1_2"}}, {"Out2", {"out2"}}}, {}));
std::shared_ptr<f::OperatorBase> grad_test_op = std::shared_ptr<f::OperatorBase> grad_test_op =
f::OpRegistry::CreateGradOp(*test_op); f::OpRegistry::CreateGradOp(*test_op);
......
...@@ -58,8 +58,8 @@ void ExposeOperator(ClassType &m) { ...@@ -58,8 +58,8 @@ void ExposeOperator(ClassType &m) {
.def("outputs", .def("outputs",
[](const typename ClassType::type &op) [](const typename ClassType::type &op)
-> std::map<std::string, std::vector<std::string>> { -> std::map<std::string, std::vector<std::string>> {
return op.outputs_; return op.outputs_;
}) })
.def("inputs", .def("inputs",
[](const typename ClassType::type &op) { return op.inputs_; }) [](const typename ClassType::type &op) { return op.inputs_; })
.def("__str__", &ClassType::type::DebugString) .def("__str__", &ClassType::type::DebugString)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册