Searched refs:a2_shape (Results 1 – 2 of 2) sorted by relevance
904 SmallVector<int64_t, 2> a2_shape(1 + block_rank * 2 + remaining_shape_rank); in convertSpaceToBatchNDOp() local907 a2_shape[0] = input_type.getShape()[0]; in convertSpaceToBatchNDOp()914 a2_shape[1 + i * 2 + 0] = padded_shape[1 + i] / block_shape_val; in convertSpaceToBatchNDOp()915 a2_shape[1 + i * 2 + 1] = block_shape_val; in convertSpaceToBatchNDOp()921 a2_shape[1 + block_rank * 2 + i] = input_shape[1 + block_rank + i]; in convertSpaceToBatchNDOp()926 RankedTensorType::get(ArrayRef<int64_t>(a2_shape), in convertSpaceToBatchNDOp()928 a1_pad_input_op.getResult(), rewriter.getI64ArrayAttr(a2_shape)); in convertSpaceToBatchNDOp()952 a3_transpose_shape[i] = a2_shape[a3_perm[i]]; in convertSpaceToBatchNDOp()2068 SmallVector<int64_t, 2> a2_shape(input_rank * 2); in convertStridedSliceOp() local2090 a2_shape[i * 2 + 0] = a1_size[i] / strides[i]; in convertStridedSliceOp()[all …]
360 vector <size_t> a2_shape = {%input.shape[0],372 …%a2_reshape = tosa.RESHAPE(%input) {new_shape=a2_shape} : (tensor<%input.type>) -> tensor<a2_shape…733 vector <size_t> a2_shape(1 + block_rank * 2 + remaining_shape_rank)734 a2_shape[0] = %input.shape[0]736 a2_shape[1 + i * 2 + 0] = %a1_pad.shape[1 + i] / block_shape.as_constant()[i]737 a2_shape[1 + i * 2 + 1] = block_shape.as_constant()[i]741 a2_shape[1 + block_rank * 2 + i] = %input.shape[1 + block_rank + i]744 …%a2_reshape = tosa.RESHAPE(%a1_pad) {new_shape=a2_shape} : (tensor<%a1_pad.type>) -> tensor<%a2_re…800 vector <size_t> a2_shape = {%input.shape[0],806 …%a2_reshape = tosa.RESHAPE(%input) {new_shape=a2_shape} : (tensor<%input.type>) -> tensor<a2_shape…[all …]