Searched refs:extended_shuffle (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dtooling_util.cc1626 std::vector<int>* extended_shuffle) {
1627 *extended_shuffle = input_shuffle;
1630 extended_shuffle->resize(newdim);
1632 (*extended_shuffle)[i] = i;
1635 (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size;
1687 std::vector<int> extended_shuffle; local
1688 ExtendShuffle(shuffle, 4, &extended_shuffle);
1699 const int input_stride_0 = input_strides[extended_shuffle[3]];
1700 const int input_stride_1 = input_strides[extended_shuffle[2]];
1701 const int input_stride_2 = input_strides[extended_shuffle[
1625 ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, std::vector<int>* extended_shuffle) argument
[all...]
H A Dtooling_util.h281 std::vector<int>* extended_shuffle);

Completed in 196 milliseconds