Searched defs:input_dim_exceeds_two (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | quantization.py | 1949 def _get_linear_node(match, input_dim_exceeds_two, input_contiguous): argument
1972 linear_node, input_index, dtype, input_dim_exceeds_two, input_contiguous argument
2008 def _is_valid_dequant_linear_pattern(dtype, input_dim_exceeds_two, input_contiguous): argument
2084 input_dim_exceeds_two=False, argument
2248 input_dim_exceeds_two=False, argument
2334 input_dim_exceeds_two=False, argument
|