Home
last modified time | relevance | path

Searched defs:channels_per_group (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DChannelShuffleKernel.cpp25 int64_t channels_per_group = channels / groups; in cpu_channel_shuffle() local
70 int64_t channels_per_group = channels / groups; in cpu_channel_shuffle_cl() local
/external/pytorch/benchmarks/operator_benchmark/pt/
Dchannel_shuffle_test.py39 def init(self, batch_size, channels_per_group, height, width, groups, channel_last): argument
/external/pytorch/aten/src/ATen/native/xnnpack/
DChannelShuffle.cpp59 int64_t channels_per_group = in channel_shuffle() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp2806 int channels_per_group = num_channels / num_groups; local
3092 int64_t channels_per_group = C / G; local