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 …]
194 array<TensorIndex, 4> kernel_shuffle; variable198 kernel_shuffle[0] = 0;199 kernel_shuffle[1] = 2;200 kernel_shuffle[2] = 3;201 kernel_shuffle[3] = 1;205 kernel_shuffle[0] = 2;206 kernel_shuffle[1] = 0;207 kernel_shuffle[2] = 1;208 kernel_shuffle[3] = 3;274 .shuffle(kernel_shuffle)[all …]