Searched refs:a2_perm (Results 1 – 2 of 2) sorted by relevance
1156 SmallVector<int32_t, 8> a2_perm(block_rank + input_rank); in convertBatchToSpaceNDOp() local1159 a2_perm[0] = block_rank; in convertBatchToSpaceNDOp()1161 a2_perm[1 + i * 2 + 0] = block_rank + 1 + i; in convertBatchToSpaceNDOp()1162 a2_perm[1 + i * 2 + 1] = i; in convertBatchToSpaceNDOp()1166 a2_perm[1 + 2 * block_rank + i] = 1 + 2 * block_rank + i; in convertBatchToSpaceNDOp()1171 a2_transpose_shape[i] = a1_shape[a2_perm[i]]; in convertBatchToSpaceNDOp()1175 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, a2_perm); in convertBatchToSpaceNDOp()
277 vector <size_t> a2_perm(%block.rank + %input.rank)279 a2_perm[0] = %block.rank281 a2_perm[1 + i * 2 + 0] = %block.rank + 1 + i282 a2_perm[1 + i * 2 + 1] = i316 …%a2_transpose = tosa.TRANSPOSE(%a1_reshape) {perms=a2_perm} : (tensor<%a1_reshape.type>) -> tensor…