“ad099595a9558106c5169ab628da83c2900db33c”上不存在“mobile/src/operators/kernel/prior_box_kernel.h”
提交 1a32189e 编写于 作者: 李寅

Merge branch 'fix-fold-reshape' into 'master'

fix fold reshape

See merge request !1137
...@@ -1192,7 +1192,11 @@ class Transformer(base_converter.ConverterInterface): ...@@ -1192,7 +1192,11 @@ class Transformer(base_converter.ConverterInterface):
and self._producer[op.input[0]].type \ and self._producer[op.input[0]].type \
== MaceOp.Reshape.name \ == MaceOp.Reshape.name \
and len(op.output_shape[0].dims) == 2: and len(op.output_shape[0].dims) == 2:
should_fold = True producer = self._producer[op.input[0]]
reshape_input_rank = len(self.get_tensor_shape(
producer.input[0]))
if reshape_input_rank == 4:
should_fold = True
if should_fold: if should_fold:
print( print(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册