get the OneHot 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)
| 1520 | |
| 1521 | @classmethod |
| 1522 | def _create_onehot(cls, onnx_node, operator, opset_version=_opset_version): |
| 1523 | """ |
| 1524 | get the OneHot operator from onnx node |
| 1525 | Args: |
| 1526 | onnx_node (OnnxNode): a given onnx node |
| 1527 | operator (Operator Class): a singa operator class |
| 1528 | opset_version (int): the opset version |
| 1529 | Returns: |
| 1530 | singa operator instance |
| 1531 | """ |
| 1532 | axis = onnx_node.getattr("axis", -1) |
| 1533 | onnx_node.set_attr_inputs(onnx_node.inputs[1], 'depth') |
| 1534 | onnx_node.set_attr_inputs(onnx_node.inputs[2], 'values') |
| 1535 | return operator(axis, None, None) |
| 1536 | |
| 1537 | @classmethod |
| 1538 | def _create_tile(cls, onnx_node, operator, opset_version=_opset_version): |
nothing calls this directly
no test coverage detected