Searched defs:input_channels_per_group (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4925 self, batch_size, input_channels_per_group, input_feature_map_shape, argument
5010 input_channels_per_group, input_feature_map_shape, argument
5147 input_channels_per_group, argument
5228 input_channels_per_group, argument
5435 input_channels_per_group, argument
5518 input_channels_per_group, argument
5775 input_channels_per_group, argument
5905 input_channels_per_group, argument
6097 input_channels_per_group, argument
6168 input_channels_per_group, argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 102 input_channels_per_group, argument
171 input_channels_per_group, argument
325 input_channels_per_group, argument
440 input_channels_per_group, argument
594 input_channels_per_group, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 761 input_channels_per_group, argument
913 input_channels_per_group, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 131 int input_channels_per_group = weight_nhwc.size(0) / num_groups; in reorder_weights_for_transpose_conv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv_prepack.cpp | 127 const int input_channels_per_group = input_channels / groups; in prepack() local
|