提交 f28625a4 编写于 作者: C chonwhite

resnet works

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