Searched refs:a3_perm (Results 1 – 2 of 2) sorted by relevance
939 SmallVector<int32_t, 8> a3_perm(a2_reshape_a1_rank); in convertSpaceToBatchNDOp() local943 a3_perm[i] = 1 + 2 * i + 1; in convertSpaceToBatchNDOp()944 a3_perm[block_rank + 1 + i] = 1 + 2 * i; in convertSpaceToBatchNDOp()946 a3_perm[block_rank] = 0; in convertSpaceToBatchNDOp()948 a3_perm[i] = i; in convertSpaceToBatchNDOp()952 a3_transpose_shape[i] = a2_shape[a3_perm[i]]; in convertSpaceToBatchNDOp()956 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, a3_perm); in convertSpaceToBatchNDOp()
753 vector <size_t> a3_perm(%a2_reshape.rank)757 a3_perm[i] = 1 + 2 * i + 1758 a3_perm[block_rank + 1 + i] = 2 * i + 1762 a3_perm[block_rank] = 0764 a3_perm[i] = i767 …%a3_reshape = tosa.RESHAPE(%a2_reshape) {perm=a3_perm} : (tensor<%a2_reshape.type>) -> tensor<%a3_…