| 2069 | } |
| 2070 | |
| 2071 | void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, |
| 2072 | std::vector<int>* extended_shuffle) { |
| 2073 | *extended_shuffle = input_shuffle; |
| 2074 | CHECK(newdim >= input_shuffle.size()); |
| 2075 | const int pad_size = newdim - input_shuffle.size(); |
| 2076 | extended_shuffle->resize(newdim); |
| 2077 | for (int i = 0; i < pad_size; i++) { |
| 2078 | (*extended_shuffle)[i] = i; |
| 2079 | } |
| 2080 | for (int i = pad_size; i < newdim; i++) { |
| 2081 | (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size; |
| 2082 | } |
| 2083 | } |
| 2084 | |
| 2085 | void ShuffleDims(const Shape& input_shape, AxesOrder input_axes_order, |
| 2086 | AxesOrder output_axes_order, Shape* output_shape) { |
no test coverage detected