Searched defs:permutation_array (Results 1 – 2 of 2) sorted by relevance
109 llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, in GetPermutationAndTransposedShape()126 llvm::ArrayRef<int64_t> permutation_array) { in GetInversePermutationArray()139 llvm::ArrayRef<int64_t> permutation_array, in GetInversePermutation()152 llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, in GetInversePermutationAndShape()520 llvm::SmallVector<int64_t, 4> permutation_array(num_spatial_dims + 2); in GetReformatTypeAndPermutation() local2813 SmallVector<int64_t, 4> permutation_array(updates_type.getRank()); in CanonicalizeScatterUpdates() local3203 SmallVector<int64_t, 4> permutation_array; in matchAndRewrite() local
40 const auto& permutation_array = model.GetArray(op->inputs[1]); in FindTransposeOpWithInput() local