From f7b63b40efe52bee8bc4a054f875e612c389dd2a Mon Sep 17 00:00:00 2001 From: Da Zheng Date: Mon, 10 Sep 2018 19:05:13 +0000 Subject: [PATCH] fix. --- src/imperative/cached_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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);