Searched refs:input_shuffle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_cuboid_convolutions.h | 498 array<TensorIndex, 5> input_shuffle; variable
502 input_shuffle = {0, 4, 1, 2, 3};
506 input_shuffle = {1, 2, 3, 0, 4};
583 input.shuffle(input_shuffle).reshape(input_dims).eval();
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2038 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle() argument
2040 *extended_shuffle = input_shuffle; in ExtendShuffle()
2041 CHECK(newdim >= input_shuffle.size()); in ExtendShuffle()
2042 const int pad_size = newdim - input_shuffle.size(); in ExtendShuffle()
2048 (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size; in ExtendShuffle()
|
| D | tooling_util.h | 285 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim,
|