Searched refs:permutation_array (Results 1 – 2 of 2) sorted by relevance
109 llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, in GetPermutationAndTransposedShape() argument111 assert(permutation_array.size() == input_type.getRank()); in GetPermutationAndTransposedShape()112 llvm::SmallVector<int64_t> transposed_shape(permutation_array.size()); in GetPermutationAndTransposedShape()113 for (int64_t i = 0; i < permutation_array.size(); ++i) { in GetPermutationAndTransposedShape()114 transposed_shape[i] = input_type.getDimSize(permutation_array[i]); in GetPermutationAndTransposedShape()119 RankedTensorType::get(permutation_array.size(), rewriter.getI64Type()), in GetPermutationAndTransposedShape()120 permutation_array); in GetPermutationAndTransposedShape()126 llvm::ArrayRef<int64_t> permutation_array) { in GetInversePermutationArray() argument128 permutation_array.size()); in GetInversePermutationArray()129 const auto permutation_array_size = permutation_array.size(); in GetInversePermutationArray()[all …]
40 const auto& permutation_array = model.GetArray(op->inputs[1]); in FindTransposeOpWithInput() local41 if (permutation_array.data_type != ArrayDataType::kInt32) { in FindTransposeOpWithInput()45 permutation_array.GetBuffer<ArrayDataType::kInt32>().data; in FindTransposeOpWithInput()