onnx_diagnostic.reference.torch_ops.controlflow_ops

class onnx_diagnostic.reference.torch_ops.controlflow_ops.If_1(node: NodeProto, version: int | None = None, parent: onnx_diagnostic.reference.TorchOnnxEvaluator | None = None)[source][source]

If

run(cond, context: Dict[str, Any] | None = None)[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.controlflow_ops.Loop_16(node: NodeProto, version: int | None = None, parent: onnx_diagnostic.reference.TorchOnnxEvaluator | None = None)[source][source]

Loop

run(M, cond, *args, context: Dict[str, Any] | None = None)[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.controlflow_ops.OpRunControlFlow(node: NodeProto, version: int | None = None, parent: onnx_diagnostic.reference.TorchOnnxEvaluator | None = None)[source][source]

Common ancestor for control flows.

classmethod has_subgraphs() bool[source][source]

Returns True if the kernel has subgraphs.