Home
last modified time | relevance | path

Searched refs:input_feature_map_size (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_functional.py36 input_feature_map_size, out_channels_per_group, groups, kernel_size, argument
41 assume(input_feature_map_size[i] + 2 * padding[i]
44 batch_size, in_channels_per_group, input_feature_map_size,
101 input_feature_map_size = (L, )
112 input_feature_map_size, out_channels_per_group, groups,
153 input_feature_map_size = (H, W)
164 input_feature_map_size, out_channels_per_group, groups,
208 input_feature_map_size = (D, H, W)
219 input_feature_map_size, out_channels_per_group, groups,
H A Dtest_quantized_module.py267 in_channels_per_group, input_feature_map_size, out_channels_per_group, argument
272 assume(input_feature_map_size[i] + 2 * padding[i]
278 batch_size, in_channels_per_group, input_feature_map_size,
451 input_feature_map_size = (length,)
478 in_channels_per_group, input_feature_map_size,
502 input_feature_map_size = (length,)
532 in_channels_per_group, input_feature_map_size,
563 input_feature_map_size = (H, W)
588 in_channels_per_group, input_feature_map_size,
616 input_feature_map_size = (H, W)
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py239 batch_size, in_channels_per_group, input_feature_map_size, argument
249 (batch_size, in_channels,) + input_feature_map_size)