Searched refs:extended_perm (Results 1 – 2 of 2) sorted by relevance
52 std::vector<int> extended_perm; in Transpose() local53 ExtendShuffle(perm, 4, &extended_perm); in Transpose()64 const int input_stride_0 = input_strides[extended_perm[3]]; in Transpose()65 const int input_stride_1 = input_strides[extended_perm[2]]; in Transpose()66 const int input_stride_2 = input_strides[extended_perm[1]]; in Transpose()67 const int input_stride_3 = input_strides[extended_perm[0]]; in Transpose()
3945 int extended_perm[4]; in Transpose() local3947 extended_perm[i] = i; in Transpose()3950 extended_perm[i + output_ext_size] = params.perm[i] + input_ext_size; in Transpose()3957 out_sizes[k] = MatchingDim(input_shape, extended_perm[k], output_shape, k); in Transpose()3964 i[extended_perm[3]] = o[3]; in Transpose()3966 i[extended_perm[2]] = o[2]; in Transpose()3968 i[extended_perm[1]] = o[1]; in Transpose()3970 i[extended_perm[0]] = o[0]; in Transpose()