Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h283 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim,
Dtooling_util.cc2082 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle() argument
2085 CHECK(newdim >= input_shuffle.size()); in ExtendShuffle()
2086 const int pad_size = newdim - input_shuffle.size(); in ExtendShuffle()
2087 extended_shuffle->resize(newdim); in ExtendShuffle()
2091 for (int i = pad_size; i < newdim; i++) { in ExtendShuffle()