get the Transpose 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)
| 1386 | |
| 1387 | @classmethod |
| 1388 | def _create_transpose(cls, |
| 1389 | onnx_node, |
| 1390 | operator, |
| 1391 | opset_version=_opset_version): |
| 1392 | """ |
| 1393 | get the Transpose operator from onnx node |
| 1394 | Args: |
| 1395 | onnx_node (OnnxNode): a given onnx node |
| 1396 | operator (Operator Class): a singa operator class |
| 1397 | opset_version (int): the opset version |
| 1398 | Returns: |
| 1399 | singa operator instance |
| 1400 | """ |
| 1401 | perm = onnx_node.getattr("perm") |
| 1402 | return operator(perm) |
| 1403 | |
| 1404 | @classmethod |
| 1405 | def _create_hardsigmoid(cls, |