MCPcopy Create free account
hub / github.com/apache/singa / _create_slice

Method _create_slice

python/singa/sonnx.py:1581–1597  ·  view source on GitHub ↗

get the Slice operator from onnx node Args: onnx_node (OnnxNode): a given onnx node operator (Operator Class): a singa operator class opset_version (int): the opset version Returns: singa operator instance

(cls, onnx_node, operator, opset_version=_opset_version)

Source from the content-addressed store, hash-verified

1579
1580 @classmethod
1581 def _create_slice(cls, onnx_node, operator, opset_version=_opset_version):
1582 """
1583 get the Slice operator from onnx node
1584 Args:
1585 onnx_node (OnnxNode): a given onnx node
1586 operator (Operator Class): a singa operator class
1587 opset_version (int): the opset version
1588 Returns:
1589 singa operator instance
1590 """
1591 onnx_node.set_attr_inputs(onnx_node.inputs[1], 'starts')
1592 onnx_node.set_attr_inputs(onnx_node.inputs[2], 'ends')
1593 if len(onnx_node.inputs) >= 4 and onnx_node.inputs[3] != '':
1594 onnx_node.set_attr_inputs(onnx_node.inputs[3], 'axes')
1595 if len(onnx_node.inputs) == 5 and onnx_node.inputs[4] != '':
1596 onnx_node.set_attr_inputs(onnx_node.inputs[4], 'steps')
1597 return operator(None, None, None, None)
1598
1599 @classmethod
1600 def _create_clip(cls, onnx_node, operator, opset_version=_opset_version):

Callers

nothing calls this directly

Calls 1

set_attr_inputsMethod · 0.80

Tested by

no test coverage detected