elasticai.creator.torch2ir.core#

Module Contents#

Classes#

Node

Implementation

Functions#

new_node

input_node

output_node

API#

class elasticai.creator.torch2ir.core.Node(name: str, data: dict[str, elasticai.creator.ir.base.Attribute])[source]#

Bases: elasticai.creator.ir.Node

implementation: str#

None

elasticai.creator.torch2ir.core.new_node(name: str, type: str, implementation: str, attributes: dict[str, Any] | None = None) → elasticai.creator.torch2ir.core.Node[source]#
elasticai.creator.torch2ir.core.input_node(attributes: dict[str, Any] | None = None) → elasticai.creator.torch2ir.core.Node[source]#
elasticai.creator.torch2ir.core.output_node(attributes: dict[str, Any] | None = None) → elasticai.creator.torch2ir.core.Node[source]#
class elasticai.creator.torch2ir.core.Implementation(*, graph: elasticai.creator.ir.GraphProtocol, name: str | None = None, type: str | None = None, data: dict[str, elasticai.creator.ir.Attribute] | None = None)[source]#

Bases: elasticai.creator.ir.Implementation[elasticai.creator.torch2ir.core.Node, elasticai.creator.ir.Edge]

name: str#

None

type: str#

None