Searched refs:extended_shuffle (Results 1 – 2 of 2) sorted by relevance
2083 std::vector<int>* extended_shuffle) { in ExtendShuffle() argument2084 *extended_shuffle = input_shuffle; in ExtendShuffle()2087 extended_shuffle->resize(newdim); in ExtendShuffle()2089 (*extended_shuffle)[i] = i; in ExtendShuffle()2092 (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size; in ExtendShuffle()2144 std::vector<int> extended_shuffle; in ShuffleArrayTemplate() local2145 ExtendShuffle(shuffle, 4, &extended_shuffle); in ShuffleArrayTemplate()2156 const int input_stride_0 = input_strides[extended_shuffle[3]]; in ShuffleArrayTemplate()2157 const int input_stride_1 = input_strides[extended_shuffle[2]]; in ShuffleArrayTemplate()2158 const int input_stride_2 = input_strides[extended_shuffle[1]]; in ShuffleArrayTemplate()[all …]
284 std::vector<int>* extended_shuffle);