ngraph.opset_utils.NodeFactory

class ngraph.opset_utils.NodeFactory(opset_version: str = 'opset6')

Bases: object

Factory front-end to create node objects.

__init__(opset_version: str = 'opset6') None

Create the NodeFactory object.

Parameters

opset_version – The opset version the factory will use to produce ops from.

Methods

__init__([opset_version])

Create the NodeFactory object.

create(op_type_name, arguments[, attributes])

Create node object from provided description.

create(op_type_name: str, arguments: List[Union[_pyngraph.Node, _pyngraph.Output]], attributes: Optional[Dict[str, Any]] = None) _pyngraph.Node

Create node object from provided description.

The user does not have to provide all node’s attributes, but only required ones.

Parameters
  • op_type_name – The operator type name.

  • arguments – The operator arguments.

  • attributes – The operator attributes.

Returns

Node object representing requested operator with attributes set.