Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc902 SmallVector<int64_t, 8> perm_values; in matchAndRewrite() local
904 perm_values.push_back(dim.getSExtValue()); in matchAndRewrite()
907 if (input_shape.size() != perm_values.size()) { in matchAndRewrite()
919 if (input_shape[perm_values[i]] != 1) { in matchAndRewrite()
920 new_major_index_ordering.push_back(perm_values[i]); in matchAndRewrite()