Searched defs:output_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 | 4926 output_channels_per_group, groups, kernels, strides, pads, dilations, argument
5011 output_channels_per_group, groups, kernels, strides, pads, o_pads, argument
5150 output_channels_per_group, argument
5231 output_channels_per_group, argument
5438 output_channels_per_group, argument
5521 output_channels_per_group, argument
5778 output_channels_per_group, argument
5909 output_channels_per_group, argument
6098 output_channels_per_group, argument
6169 output_channels_per_group, argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 105 output_channels_per_group, argument
174 output_channels_per_group, argument
328 output_channels_per_group, argument
443 output_channels_per_group, argument
597 output_channels_per_group, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 764 output_channels_per_group, argument
916 output_channels_per_group, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 132 int output_channels_per_group = weight_nhwc.size(1); in reorder_weights_for_transpose_conv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv_prepack.cpp | 129 const int output_channels_per_group = output_channels / groups; in prepack() local
|