Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
clean code
Browse files Browse the repository at this point in the history
  • Loading branch information
wuxun-zhang committed Mar 25, 2020
1 parent 3bb434f commit bdb94de
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions src/operator/nn/pooling.cc
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,6 @@ void PoolingComputeExCPU(const nnvm::NodeAttrs &attrs, const OpContext &ctx,

// Pooling does not currently support working with views
if (inputs[0].IsView() || outputs[0].IsView()) {
std::cout << "Fall back to Pooling backward pass..." << std::endl;
FallBackCompute(PoolingCompute<cpu>, attrs, ctx, inputs, req, outputs);
return;
}
Expand All @@ -289,7 +288,6 @@ void PoolingComputeExCPU(const nnvm::NodeAttrs &attrs, const OpContext &ctx,
MKLDNN_OPCHECK_RUN(PoolingCompute<cpu>, attrs, ctx, inputs, req, outputs);
return;
}
std::cout << "Fall back to Pooling forward pass..." << std::endl;
FallBackCompute(PoolingCompute<cpu>, attrs, ctx, inputs, req, outputs);
}

Expand All @@ -301,7 +299,6 @@ void PoolingGradComputeExCPU(const nnvm::NodeAttrs &attrs, const OpContext &ctx,

// Pooling does not currently support working with views
if (inputs[0].IsView() || outputs[0].IsView()) {
std::cout << "Fall back to Pooling backward pass..." << std::endl;
FallBackCompute(PoolingGradCompute<cpu>, attrs, ctx, inputs, req, outputs);
return;
}
Expand Down Expand Up @@ -330,7 +327,6 @@ void PoolingGradComputeExCPU(const nnvm::NodeAttrs &attrs, const OpContext &ctx,
outputs);
return;
}
std::cout << "Fall back to Pooling backward pass..." << std::endl;
FallBackCompute(PoolingGradCompute<cpu>, attrs, ctx, inputs, req, outputs);
}

Expand Down

0 comments on commit bdb94de

Please sign in to comment.