提交 badf833e 编写于 作者: L liuqi

Fix bug: the output shape is not right in benchmark.

上级 382350bb
...@@ -37,7 +37,7 @@ Operator lists ...@@ -37,7 +37,7 @@ Operator lists
"RELU1","Y","" "RELU1","Y",""
"RELU6","Y","" "RELU6","Y",""
"RELUX","Y","" "RELUX","Y",""
"RESHAPE","Y","Limited support: GPU is full supported, for CPU only supports softmax-like usage." "RESHAPE","Y","Limited support: GPU only supports softmax-like usage, CPU only supports the usage which not change the storage format."
"RESIZE_BILINEAR","Y","" "RESIZE_BILINEAR","Y",""
"RNN","","" "RNN","",""
"RPN_PROPOSAL_LAYER","Y","" "RPN_PROPOSAL_LAYER","Y",""
......
...@@ -114,9 +114,8 @@ MaceStatus SerialNet::Run(RunMetadata *run_metadata) { ...@@ -114,9 +114,8 @@ MaceStatus SerialNet::Run(RunMetadata *run_metadata) {
} }
std::vector<std::vector<int64_t>> output_shapes; std::vector<std::vector<int64_t>> output_shapes;
for (auto output_shape : op->debug_def().output_shape()) { for (auto output : op->Outputs()) {
output_shapes.push_back({output_shape.dims().begin(), output_shapes.push_back(output->shape());
output_shape.dims().end()});
} }
OperatorStats op_stats = {op->debug_def().name(), op->debug_def().type(), OperatorStats op_stats = {op->debug_def().name(), op->debug_def().type(),
output_shapes, output_shapes,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册