操作类型

用于分类onnx操作的实用函数。

函数

is_binary_op	返回给定的操作符是否为二元操作符。
is_control_flow_op	返回给定的操作类型是否属于控制流类别。
is_conversion_op	返回给定的操作类型是否属于转换类别。
is_copy_op	返回给定的操作是否是复制操作符。
is_default_quantizable_op_by_ort	返回ORT是否默认量化操作类型。
is_fusible_reduction_op	返回给定的操作类型是否属于归约类别并且可由编译器融合。
is_fusible_scaling_op	返回给定的操作类型是否属于缩放类别并且可以与输入融合。
is_generator_op	返回给定的操作类型是否属于生成器类别。
is_irregular_mem_access_op	返回给定的操作类型是否属于不规则内存访问类别。
is_linear_op	返回给定的操作类型是否属于线性类别。
is_modifier_op	返回给定的操作类型是否属于Modifier类别。
is_multiclass_op	返回给定的操作类型是否属于多类别。
is_non_reshape_copy_op	返回给定的操作是否是一个非重塑的复制操作。
is_normalization_op	返回给定的操作类型是否属于归一化类别。
is_pointwise_or_elementwise_op	返回给定的操作类型是否属于点对点或元素对元素类别。
is_pooling_or_window_op	返回给定的操作类型是否属于池化/窗口类别。
is_recurrent_op	返回给定的操作类型是否属于循环类别。
is_selection_op	返回给定的操作类型是否属于选择类别。
is_sequence_op	返回给定的操作类型是否属于序列类别。
is_shape_op	返回给定的操作类型是否属于形状类别。
is_unary_op	返回给定的操作符是否为一元操作符。

is_binary_op(op_type)

返回给定的操作符是否为二元操作符。

Parameters:

op_type (str) –

is_control_flow_op(op_type)

返回给定的操作类型是否属于控制流类别。

Parameters:

op_type (str) –

is_conversion_op(op_type)

返回给定的操作类型是否属于转换类别。

Parameters:

op_type (str) –

is_copy_op(op_type)

返回给定的操作是否是复制操作符。

Parameters:

op_type (str) –

is_default_quantizable_op_by_ort(op_type)

返回ORT是否默认量化操作类型。

注意。可能会因不同的ORT版本而有所变化。 注意。用户可以使用nodes_to_quantize和/或op_types_to_quantize参数来量化非默认操作。 参考：https://github.com/microsoft/onnxruntime/blob/main/onnxruntime/python/tools/quantization/registry.py

Parameters:

op_type (str) –

is_fusible_reduction_op(op_type)

返回给定的操作类型是否属于归约类别并且可由编译器融合。

Parameters:

op_type (str) –

is_fusible_scaling_op(op_type)

返回给定的操作类型是否属于缩放类别并且可以与输入融合。

Parameters:

op_type (str) –

is_generator_op(op_type)

返回给定的操作类型是否属于生成器类别。

Parameters:

op_type (str) –

is_irregular_mem_access_op(op_type)

返回给定的操作类型是否属于不规则内存访问类别。

Parameters:

op_type (str) –

is_linear_op(op_type)

返回给定的操作类型是否属于线性类别。

Parameters:

op_type (str) –

is_modifier_op(op_type)

返回给定的操作类型是否属于Modifier类别。

Parameters:

op_type (str) –

is_multiclass_op(op_type)

返回给定的操作类型是否属于多类别。

Parameters:

op_type (str) –

is_non_reshape_copy_op(op_type)

返回给定的操作是否是一个非重塑的复制操作。

Parameters:

op_type (str) –

is_normalization_op(op_type)

返回给定的操作类型是否属于归一化类别。

Parameters:

op_type (str) –

is_pointwise_or_elementwise_op(op_type)

返回给定的操作类型是否属于点对点或元素对元素类别。

这仅考虑可融合的类型。

Parameters:

op_type (str) –

is_pooling_or_window_op(op_type)

返回给定的操作类型是否属于池化/窗口类别。

Parameters:

op_type (str) –

is_recurrent_op(op_type)

返回给定的操作类型是否属于循环类别。

Parameters:

op_type (str) –

is_selection_op(op_type)

返回给定的操作类型是否属于选择类别。

Parameters:

op_type (str) –

is_sequence_op(op_type)

返回给定的操作类型是否属于序列类别。

Parameters:

op_type (str) –

is_shape_op(op_type)

返回给定的操作类型是否属于形状类别。

Parameters:

op_type (str) –

is_unary_op(op_type)

返回给定的操作符是否为一元操作符。

Parameters:

op_type (str) –