提交 f28625a4 编写于 作者: C chonwhite

resnet works

上级 8e699af1
...@@ -73,7 +73,6 @@ class Debugger { ...@@ -73,7 +73,6 @@ class Debugger {
// op_config["nms"] = true; // op_config["nms"] = true;
// op_config["pb_boxes"] = true; // op_config["pb_boxes"] = true;
// op_config["pb_variances"] = true; // op_config["pb_variances"] = true;
// op_config["softmax"] = true; // op_config["softmax"] = true;
} }
}; };
......
...@@ -40,7 +40,6 @@ void ElementwiseAddCompute::PrepareForRun() { ...@@ -40,7 +40,6 @@ void ElementwiseAddCompute::PrepareForRun() {
pe_.apply(); pe_.apply();
} }
void ElementwiseAddCompute::Run() { void ElementwiseAddCompute::Run() {
usleep(50 * 100 * 1000);
pe_.dispatch(); pe_.dispatch();
#ifdef FPGA_PRINT_TENSOR #ifdef FPGA_PRINT_TENSOR
zynqmp::ElementwiseAddParam& ew_param = pe_.param(); zynqmp::ElementwiseAddParam& ew_param = pe_.param();
...@@ -63,7 +62,6 @@ void ElementwiseAddActivationCompute::PrepareForRun() { ...@@ -63,7 +62,6 @@ void ElementwiseAddActivationCompute::PrepareForRun() {
pe_.apply(); pe_.apply();
} }
void ElementwiseAddActivationCompute::Run() { void ElementwiseAddActivationCompute::Run() {
usleep(500 * 100 * 1000);
pe_.dispatch(); pe_.dispatch();
#ifdef FPGA_PRINT_TENSOR #ifdef FPGA_PRINT_TENSOR
zynqmp::ElementwiseAddParam& ew_param = pe_.param(); zynqmp::ElementwiseAddParam& ew_param = pe_.param();
......
...@@ -41,15 +41,12 @@ void ReshapeCompute::Run() { ...@@ -41,15 +41,12 @@ void ReshapeCompute::Run() {
output_dims = lite::operators::ValidateShape(shape, x_dims); output_dims = lite::operators::ValidateShape(shape, x_dims);
output->Resize(output_dims); output->Resize(output_dims);
} }
if (inplace) { // if (inplace) {
output->ShareDataWith(*x); // output->ShareDataWith(*x);
} else { // } else {
output->CopyDataFrom(*x); // output->CopyDataFrom(*x);
} // }
output->ZynqTensor()->copyFrom(x->ZynqTensor());
param.x->ZynqTensor()->saveToFile("reshape_in", true);
output->ZynqTensor()->saveToFile("reshape_out", true);
output->Resize(output_dims); output->Resize(output_dims);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册