- elif op_name in convert_map:
- sym = convert_map[op_name](inputs, attrs, self._params)
- elif op_name in convert_map_rnn:
- sym = self._convert_rnn_operator(op_name, inputs, attrs,
- self._params, graph,
- convert_map_rnn)
- else:
- raise NotImplementedError("Operator {} not implemented.".format(op_name))
- return sym
-
- def _fix_extranodes(self, op_name, attr, inputs):
- if op_name == "Softmax":
- # Require some times flatten of data before it goes to softmax
- # Need to relook into this with latest softmax axis support.
- op = AttrCvt(op_name='flatten')(inputs, {})
- node_output = op.list_output_names()
- for k, i in zip(list(node_output), range(len(node_output))):
- self._nodes[k] = op[i]
- inputs = [op]
-
- return inputs
