diff --git a/src/imperative/cached_op.cc b/src/imperative/cached_op.cc index a7218e9bca64..bda9e750caed 100644 --- a/src/imperative/cached_op.cc +++ b/src/imperative/cached_op.cc @@ -1339,7 +1339,7 @@ NNVM_REGISTER_OP(_CachedOp) const CachedOpPtr& op = nnvm::get(attrs.parsed); return op::DefaultSubgraphOpResourceRequest1(op->GetForwardSym()); }) -.set_attr("FExecType", DefaultSubgraphOpExecType) +.set_attr("FExecType", op::DefaultSubgraphOpExecType) .add_argument("data", "NDArray-or-Symbol[]", "input data list"); NNVM_REGISTER_OP(_backward_CachedOp) @@ -1361,7 +1361,7 @@ NNVM_REGISTER_OP(_backward_CachedOp) }) .set_attr("FStatefulComputeEx", CachedOpBackward) .set_attr("FStatefulComputeEx", CachedOpBackward) -.set_attr("FExecType", DefaultSubgraphOpExecType) +.set_attr("FExecType", op::DefaultSubgraphOpExecType) .set_attr("TIsLayerOpBackward", true) .set_attr("TIsBackward", true);