Searched refs:input_shuffle (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 503 array<TensorIndex, 4> input_shuffle; variable
507 input_shuffle = {0, 3, 1, 2};
511 input_shuffle = {1, 2, 0, 3};
584 input.shuffle(input_shuffle).eval().reshape(input_dims);
|
| 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.h | 283 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim,
|
| D | tooling_util.cc | 2082 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle() argument
2084 *extended_shuffle = input_shuffle; in ExtendShuffle()
2085 CHECK(newdim >= input_shuffle.size()); in ExtendShuffle()
2086 const int pad_size = newdim - input_shuffle.size(); in ExtendShuffle()
2092 (*extended_shuffle)[i] = input_shuffle[i - pad_size] + pad_size; in ExtendShuffle()
|