Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc978 SmallVector<int64_t, 6> permutation_vals; in matchAndRewrite() local
980 permutation_vals.push_back(2 + 2 * i); in matchAndRewrite()
982 permutation_vals.push_back(0); in matchAndRewrite()
984 permutation_vals.push_back(1 + 2 * i); in matchAndRewrite()
987 permutation_vals.push_back(block_rank + i); in matchAndRewrite()
990 loc, GetI64ElementsAttr(permutation_vals, &rewriter)); in matchAndRewrite()