Searched refs:a1_shape (Results 1 – 2 of 2) sorted by relevance
1133 SmallVector<int64_t, 2> a1_shape(block_rank + input_rank); in convertBatchToSpaceNDOp() local1135 for (int i = 0; i < block_rank; i++) a1_shape[i] = block_shape[i]; in convertBatchToSpaceNDOp()1137 a1_shape[block_rank] = input_shape[0] / block_num_elems; in convertBatchToSpaceNDOp()1140 a1_shape[i + block_rank + 1] = input_shape[i + 1]; in convertBatchToSpaceNDOp()1144 RankedTensorType::get(ArrayRef<int64_t>(a1_shape), in convertBatchToSpaceNDOp()1146 input_value, rewriter.getI64ArrayAttr(a1_shape)); in convertBatchToSpaceNDOp()1171 a2_transpose_shape[i] = a1_shape[a2_perm[i]]; in convertBatchToSpaceNDOp()
259 vector <size_t> a1_shape(%block.rank + %input.rank)262 a1_shape[i] = %block.shape[i]265 a1_shape[%block.rank] = %input.shape.[0] / %block.num_elements268 a1_shape[i + %block.rank] = %input.shape[i]315 …%a1_reshape = tosa.RESHAPE(%input) {new_shape=a1_shape} : (tensor<%input.type>) -> tensor<a1_shape…