afe.backends.apu.apu_runner

Classes

ARMBackendRunner

Implementation of execution on an ARM processor,

Functions

is_arm_backend_node(→ bool)

Test whether the node is compiled backend code for ARM.

Module Contents

afe.backends.apu.apu_runner.is_arm_backend_node(node: afe.ir.node.AwesomeNode) → bool[source]

Test whether the node is compiled backend code for ARM.

class afe.backends.apu.apu_runner.ARMBackendRunner(host: str, port: int)[source]

Implementation of execution on an ARM processor, conforming to the protocol of AwesomeNet.run.

execute_node(node: afe.ir.node.AwesomeNode, inputs: Dict[afe.ir.defines.InputName, afe.ir.sima_ir.SiMaIRTensorTypes], node_outputs: Dict[afe.ir.defines.NodeName, numpy.ndarray | Tuple[numpy.ndarray, Ellipsis]])[source]