From 72b1b690d444f062db7adb2eeaa2527a49076d66 Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Tue, 18 Sep 2018 17:16:26 +0800 Subject: [PATCH] add fetch and feed op for tools --- python/tools/mdl2fluid/mdl2fluid.py | 34 +++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/python/tools/mdl2fluid/mdl2fluid.py b/python/tools/mdl2fluid/mdl2fluid.py index 504b4403cc..d6a1f1cfd9 100644 --- a/python/tools/mdl2fluid/mdl2fluid.py +++ b/python/tools/mdl2fluid/mdl2fluid.py @@ -34,6 +34,9 @@ class Converter: print self.program_desc.blocks def package_ops(self, block_desc): + + self.add_op_feed(block_desc) + # add ops with layer if 'layer' in self.mdl_json: @@ -58,6 +61,37 @@ class Converter: self.package_ops_inputs(desc_ops_add, layer) self.package_ops_attrs(desc_ops_add, layer) + self.add_op_fetch(block_desc) + + def add_op_feed(self, block_desc): + desc_ops_add = block_desc.ops.add() + inputs_add = desc_ops_add.inputs.add() + inputs_add.parameter = 'X' + inputs_add.arguments.append('feed') + desc_ops_add.type = 'feed' + outputs_add = desc_ops_add.outputs.add() + outputs_add.parameter = 'Out' + outputs_add.arguments.append('data') + attrs_add = desc_ops_add.attrs.add() + attrs_add.name = 'col' + # boolean + attrs_add.type = 0 + attrs_add.i = 0 + + def add_op_fetch(self, block_desc): + desc_ops_add = block_desc.ops.add() + inputs_add = desc_ops_add.inputs.add() + inputs_add.parameter = 'X' + inputs_add.arguments.append('conv_pred_87') + desc_ops_add.type = 'fetch' + outputs_add = desc_ops_add.outputs.add() + outputs_add.parameter = 'Out' + outputs_add.arguments.append('fetch') + attrs_add = desc_ops_add.attrs.add() + attrs_add.name = 'col' + # boolean + attrs_add.type = 0 + attrs_add.i = 0 @staticmethod def package_ops_attrs(desc_ops_add, layer): -- GitLab