Home
last modified time | relevance | path

Searched refs:_set_output_shape (Results 1 – 4 of 4) sorted by relevance

/dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/paddle/
H A Dpaddle_parser.py128 def _set_output_shape(source_node, IR_node, output_shapes): member in PaddleParser
239 PaddleParser._set_output_shape(source_node, IR_node, output_shapes)
306 PaddleParser._set_output_shape(source_node, IR_node_act, output_shapes)
332 PaddleParser._set_output_shape(source_node, IR_node, output_shapes)
393 PaddleParser._set_output_shape(source_node, IR_node_act, output_shapes)
440 PaddleParser._set_output_shape(source_node, IR_node, output_shapes)
472 PaddleParser._set_output_shape(source_node, IR_node_act, output_shapes)
495 PaddleParser._set_output_shape(source_node, IR_node, output_shapes)
534 PaddleParser._set_output_shape(source_node, IR_node_act, output_shapes)
590 PaddleParser._set_output_shape(source_node, IR_node, output_shapes)
[all …]
/dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/keras/
H A Dkeras2_parser.py156 def _set_output_shape(source_node, IR_node): member in Keras2Parser
173 Keras2Parser._set_output_shape(source_node, IR_node)
224 Keras2Parser._set_output_shape(source_node, IR_node)
351 Keras2Parser._set_output_shape(source_node, flatten_node)
/dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/pytorch/
H A Dpytorch_parser.py114 def _set_output_shape(self, source_node, IR_node): member in PytorchParser
488 self._set_output_shape(source_node, IR_node)
/dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/coreml/
H A Dcoreml_parser.py158 def _set_output_shape(source_node, IR_node): member in CoremlParser
217 CoremlParser._set_output_shape(source_node, IR_node)