Home
last modified time | relevance | path

Searched defs:input_shuffle (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions.h484 array<TensorIndex, 4> input_shuffle; variable
H A Deigen_backward_cuboid_convolutions.h498 array<TensorIndex, 5> input_shuffle; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc2084 void ExtendShuffle(const std::vector<int>& input_shuffle, int newdim, in ExtendShuffle()