Searched refs:extended_shuffle (Results 1 – 2 of 2) sorted by relevance
2039 std::vector<int>* extended_shuffle) { in ExtendShuffle() argument2040 *extended_shuffle = input_shuffle; in ExtendShuffle()2043 extended_shuffle->resize(newdim); in ExtendShuffle()2045 (*extended_shuffle)[i] = i; in ExtendShuffle()2048 (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size; in ExtendShuffle()2100 std::vector<int> extended_shuffle; in ShuffleArrayTemplate() local2101 ExtendShuffle(shuffle, 4, &extended_shuffle); in ShuffleArrayTemplate()2112 const int input_stride_0 = input_strides[extended_shuffle[3]]; in ShuffleArrayTemplate()2113 const int input_stride_1 = input_strides[extended_shuffle[2]]; in ShuffleArrayTemplate()2114 const int input_stride_2 = input_strides[extended_shuffle[1]]; in ShuffleArrayTemplate()[all …]
286 std::vector<int>* extended_shuffle);