Searched refs:permutation_tensor (Results 1 – 1 of 1) sorted by relevance
2138 Tensor permutation_tensor; in RemoveShuffleOrTranspose() local2139 if (GetTensorFromConstNode(node->input(1), &permutation_tensor) && in RemoveShuffleOrTranspose()2143 for (int j = 0; j < permutation_tensor.NumElements(); ++j) { in RemoveShuffleOrTranspose()2144 if (permutation_tensor.dtype() == DT_INT64) { in RemoveShuffleOrTranspose()2145 permutation.push_back(permutation_tensor.vec<int64>()(j)); in RemoveShuffleOrTranspose()2147 permutation.push_back(permutation_tensor.vec<int>()(j)); in RemoveShuffleOrTranspose()