Fix error

8 years ago · e4f058cec7
parent 54fad18382
commit e4f058cec7
1 changed files with 1 additions and 1 deletions
--- a/paddle/framework/pybind.cc
+++ b/paddle/framework/pybind.cc
@ -29,7 +29,7 @@ limitations under the License. */
 namespace py = pybind11;

 USE_OP(add_two);
-USE_OP_CPU(onehot_cross_entropy);
+USE_CPU_OP(onehot_cross_entropy);
 USE_OP_ITSELF(fc);
 USE_OP(sgd);
 USE_OP(mul);