Searched refs:a3_reshape_dims (Results 1 – 1 of 1) sorted by relevance
1741 SmallVector<int64_t, 4> a3_reshape_dims; in convertSpaceToDepthOp() local1742 a3_reshape_dims.push_back(input_shape[0]); in convertSpaceToDepthOp()1743 a3_reshape_dims.push_back(input_shape[1] / block_size[0]); in convertSpaceToDepthOp()1744 a3_reshape_dims.push_back(input_shape[2] / block_size[1]); in convertSpaceToDepthOp()1745 a3_reshape_dims.push_back(input_shape[3] * block_size[0] * block_size[1]); in convertSpaceToDepthOp()1748 ArrayRef<int64_t>(a3_reshape_dims), output_type.getElementType()); in convertSpaceToDepthOp()1752 rewriter.getI64ArrayAttr(a3_reshape_dims)) in convertSpaceToDepthOp()1826 SmallVector<int64_t, 4> a3_reshape_dims; in convertDepthToSpaceOp() local1827 a3_reshape_dims.push_back(input_shape[0]); in convertDepthToSpaceOp()1828 a3_reshape_dims.push_back(input_shape[1] * block_size[0]); in convertDepthToSpaceOp()[all …]