Home
last modified time | relevance | path

Searched refs:dim_info (Results 1 - 2 of 2) sorted by relevance

/third_party/mindspore/test/utils/
H A Dcommon.cpp27 * dim_info: array to store the info of the dim in datafile, like
31 void getDimInfo(FILE *fp, std::vector<int64_t>* dim_info) { in getDimInfo() argument
39 dim_info->push_back(*dim_buffer); // get dim size in getDimInfo()
44 for (; i <= dim_info->at(0); i++) { in getDimInfo()
45 dim_info->push_back(*(dim_buffer + i)); in getDimInfo()
48 dim_info->push_back(data_size); in getDimInfo()
58 * dim_info: array to store the info of the dim in datafile, like [4,3,3,6,3],4
70 std::vector<int64_t> dim_info; in readTestDataFile() local
71 std::vector<int64_t>* ptr_dim_info = &dim_info; in readTestDataFile()
81 uint32_t *memory = reinterpret_cast<uint32_t *>(malloc((dim_info[ in readTestDataFile()
318 std::vector<int64_t> dim_info; compFp32WithTData() local
346 std::vector<int64_t> dim_info; compUint8WithTData() local
[all...]
H A Dcommon.h37 void getDimInfo(FILE *fp, std::vector<int64_t>* dim_info);

Completed in 2 milliseconds