Lines Matching full:permute
24 #include "Permute.h"
38 SimpleTensor<T> permute(const SimpleTensor<T> &src, PermutationVector perm) in permute() function
40 // Permute shapes in permute()
42 permute(dst_shape, perm); in permute()
53 permute(dst_coords, perm); in permute()
61 template SimpleTensor<int8_t> permute(const SimpleTensor<int8_t> &src, PermutationVector perm);
62 template SimpleTensor<uint8_t> permute(const SimpleTensor<uint8_t> &src, PermutationVector perm);
63 template SimpleTensor<int16_t> permute(const SimpleTensor<int16_t> &src, PermutationVector perm);
64 template SimpleTensor<uint16_t> permute(const SimpleTensor<uint16_t> &src, PermutationVector perm);
65 template SimpleTensor<uint32_t> permute(const SimpleTensor<uint32_t> &src, PermutationVector perm);
66 template SimpleTensor<float> permute(const SimpleTensor<float> &src, PermutationVector perm);
67 template SimpleTensor<half> permute(const SimpleTensor<half> &src, PermutationVector perm);