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, verbose: int = 0)

If

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

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, verbose: int = 0)

Loop

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

Kernel implementation.

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

Common ancestor for control flows.

classmethod has_subgraphs() → bool

Returns True if the kernel has subgraphs.