Searched refs:kernel_shuffle (Results 1 – 2 of 2) sorted by relevance
190 array<TensorIndex, 5> kernel_shuffle; variable194 kernel_shuffle[0] = 0;195 kernel_shuffle[1] = 2;196 kernel_shuffle[2] = 3;197 kernel_shuffle[3] = 4;198 kernel_shuffle[4] = 1;202 kernel_shuffle[0] = 3;203 kernel_shuffle[1] = 0;204 kernel_shuffle[2] = 1;205 kernel_shuffle[3] = 2;[all …]
195 array<TensorIndex, 4> kernel_shuffle; variable199 kernel_shuffle[0] = 0;200 kernel_shuffle[1] = 2;201 kernel_shuffle[2] = 3;202 kernel_shuffle[3] = 1;206 kernel_shuffle[0] = 2;207 kernel_shuffle[1] = 0;208 kernel_shuffle[2] = 1;209 kernel_shuffle[3] = 3;280 .shuffle(kernel_shuffle)[all …]