| | __init__.py | 3.6 KB |
| | _debug_passes.py | 840 B |
| | annotate_channels_last_dim_order_pass.py | 9.0 KB |
| | annotate_decomposed_matmul.py | 4.8 KB |
| | arm_pass_manager.py | 9.3 KB |
| | arm_pass_utils.py | 7.7 KB |
| | arm_pass.py | 1.1 KB |
| | broadcast_args_pass.py | 2.2 KB |
| | cast_int64_pass.py | 2.4 KB |
| | cast_to_int32_pass.py | 1.6 KB |
| | conv1d_unsqueeze_pass.py | 6.0 KB |
| | convert_any_default_dim_dims_pass.py | 3.5 KB |
| | convert_expand_copy_to_repeat.py | 1.9 KB |
| | convert_full_like_to_full_pass.py | 1.1 KB |
| | convert_minmax_pass.py | 5.1 KB |
| | convert_split_to_slice.py | 2.7 KB |
| | convert_squeezes_to_view.py | 1.0 KB |
| | convert_to_clamp.py | 1.1 KB |
| | decompose_batchnorm_pass.py | 5.0 KB |
| | decompose_cosine_similarity_pass.py | 2.6 KB |
| | decompose_div_pass.py | 1.6 KB |
| | decompose_gelu_pass.py | 5.6 KB |
| | decompose_layernorm_pass.py | 6.8 KB |
| | decompose_leaky_relu_pass.py | 2.3 KB |
| | decompose_linalg_vector_norm_pass.py | 2.9 KB |
| | decompose_linear_pass.py | 4.7 KB |
| | decompose_meandim_pass.py | 5.0 KB |
| | decompose_ne_pass.py | 2.3 KB |
| | decompose_select.py | 2.0 KB |
| | decompose_silu_pass.py | 978 B |
| | decompose_softmax_pass.py | 2.8 KB |
| | decompose_softmax_unstable_pass.py | 2.4 KB |
| | decompose_sqrt_pass.py | 1.2 KB |
| | decompose_sum_pass.py | 2.6 KB |
| | decompose_var_pass.py | 3.1 KB |
| | fold_qdq_with_annotated_qparams_pass.py | 10.0 KB |
| | fuse_batchnorm2d_pass.py | 6.5 KB |
| | fuse_constant_ops_pass.py | 7.8 KB |
| | fuse_equal_placeholders_pass.py | 2.9 KB |
| | fuse_quantized_activation_pass.py | 2.1 KB |
| | insert_rescales_pass.py | 4.3 KB |
| | insert_table_ops.py | 11.7 KB |
| | match_arg_ranks_pass.py | 3.3 KB |
| | match_where_self_arg_dtype_pass.py | 3.7 KB |
| | mm_to_bmm_pass.py | 3.8 KB |
| | remove_clone_pass.py | 642 B |
| | replace_inf_values_pass.py | 1.6 KB |
| | replace_scalar_with_tensor_pass.py | 2.5 KB |
| | scalars_to_attribute_pass.py | 3.5 KB |
| | size_adjust_conv2d_pass.py | 6.6 KB |
| | TARGETS | 659 B |
| | unsqueeze_before_repeat_pass.py | 2.2 KB |
| | unsqueeze_scalar_placeholders_pass.py | 2.3 KB |
