diff --git a/paddle/framework/pybind.cc b/paddle/framework/pybind.cc index 915ffb1c007c24eca457a0c646378bc191a15810..cc2bad6234fd77c631617dc1cedef0465bbe7840 100644 --- a/paddle/framework/pybind.cc +++ b/paddle/framework/pybind.cc @@ -30,7 +30,7 @@ namespace py = pybind11; USE_OP(add_two); USE_OP_CPU(onehot_cross_entropy); -USE_OP_WITHOUT_KERNEL(fc); +USE_OP_ITSELF(fc); USE_OP(sgd); USE_OP(mul); USE_OP(mean); @@ -38,7 +38,7 @@ USE_OP(sigmoid); USE_OP(softmax); USE_OP(rowwise_add); USE_OP(fill_zeros_like); -USE_OP_WITHOUT_KERNEL(recurrent_op); +USE_OP_ITSELF(recurrent_op); USE_OP(uniform_random); namespace paddle { namespace framework { diff --git a/paddle/operators/recurrent_op_test.cc b/paddle/operators/recurrent_op_test.cc index 6ce28a2b52e3b90596a68714bfdbc07d2d4814d5..0c9a343415835540c7543f15f40c53b78a6a55c4 100644 --- a/paddle/operators/recurrent_op_test.cc +++ b/paddle/operators/recurrent_op_test.cc @@ -395,4 +395,4 @@ TEST(RecurrentOp, LinkMemories) { USE_OP(add_two); USE_OP(mul); -USE_OP_WITHOUT_KERNEL(recurrent_op); +USE_OP_ITSELF(recurrent_op);