experimental_experiment.xbuilder.type_inference

experimental_experiment.xbuilder.type_inference.infer_types(node: NodeProto, input_types: Sequence[int], output_name: str | None, exc: bool = True) → int | Tuple[int]

Tries to infer the type of an output or all outputs.

Parameters:

• node – NodeProto

• input_types – type of the elements of the input tensors

• output_name – type for the desired output or all types if all are needed

• exc – raise an exception if type cannot be infered

Returns:

tuple of types or output type