Searched defs:input_shuffle (Results 1 – 2 of 2) sorted by relevance
498 array<TensorIndex, 5> input_shuffle; variable
2038 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle()