Home
last modified time | relevance | path

Searched refs:newdim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2038 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle() argument
2041 CHECK(newdim >= input_shuffle.size()); in ExtendShuffle()
2042 const int pad_size = newdim - input_shuffle.size(); in ExtendShuffle()
2043 extended_shuffle->resize(newdim); in ExtendShuffle()
2047 for (int i = pad_size; i < newdim; i++) { in ExtendShuffle()
Dtooling_util.h285 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim,