get the ScatterElements from the 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)
| 1709 | |
| 1710 | @classmethod |
| 1711 | def _create_scatter_elements(cls, |
| 1712 | onnx_node, |
| 1713 | operator, |
| 1714 | opset_version=_opset_version): |
| 1715 | """ |
| 1716 | get the ScatterElements from the onnx node |
| 1717 | Args: |
| 1718 | onnx_node(OnnxNode): a given onnx node |
| 1719 | operator (Operator Class): a singa operator class |
| 1720 | opset_version(int): the opset version |
| 1721 | Returns: |
| 1722 | singa operator instance |
| 1723 | """ |
| 1724 | axis = onnx_node.getattr("axis", 0) |
| 1725 | onnx_node.set_attr_inputs(onnx_node.inputs[1], 'indices') |
| 1726 | onnx_node.set_attr_inputs(onnx_node.inputs[2], 'updates') |
| 1727 | return operator(None, None, axis) |
| 1728 | |
| 1729 | @classmethod |
| 1730 | def _onnx_constant_to_np(cls, onnx_node, opset_version=_opset_version): |
nothing calls this directly
no test coverage detected