forked from RUSH-LAB/Flash
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdataset.h
15 lines (10 loc) · 781 Bytes
/
dataset.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#pragma once
using namespace std;
void anshuReadSparse(string fileName, int *indices, int *markers, unsigned int n, unsigned int bufferlen);
void readSparse(string fileName, int offset, int n, int *indices, float *values, int *markers, unsigned int bufferlen);
void Read_MNIST(string fileName, int NumberOfImages, int DataOfAnImage, float *data);
void Read_MNISTLabel(string fileName, int numToRead, int *labels);
void fvecs_read(const std::string& file, int offset, int readsize, float *out);
void bvecs_read(const std::string& file, int offset, int readsize, float *out);
void readGroundTruthInt(const std::string& file, int numQueries, int availableTopK, unsigned int *out);
void readGroundTruthFloat(const std::string& file, int numQueries, int availableTopK, float *out);