Searched refs:nz_cnt (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | graph_util.h | 166 … size_t nz_cnt, size_t coor_best_bit, size_t bit_num, schema::TensorT *tensor) { in SparsityCompress() argument
183 bits[index++] = (nz_cnt >> (32 - i - 1)) & 0x1; in SparsityCompress()
196 std::vector<size_t> coors(nz_cnt); in SparsityCompress()
204 MS_ASSERT(coors_index < nz_cnt); in SparsityCompress()
297 …auto nz_cnt = CalCoorBestBit<T>(quant_data, elem_cnt, quant_params, unique_value_bit, &coor_best_b… in PackRepetition() local
300 …1 * 8 + 4 * 8 + bit_num + bit_num * unique_value_cnt + unique_value_bit * nz_cnt + nz_cnt * coor_b… in PackRepetition()
314 pack_sparsity_size_in_byte, nz_cnt, coor_best_bit, bit_num, tensor); in PackRepetition()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | weight_decoder.cc | 117 size_t nz_cnt = 0; in SparseDecompress() local
120 nz_cnt |= bit << static_cast<size_t>((kBit32 - i - 1)); in SparseDecompress()
146 for (size_t i = 0; i < nz_cnt; i++) { in SparseDecompress()
157 for (size_t i = 0; i < nz_cnt; i++) { in SparseDecompress()
|
| D | weight_decoder.h | 73 auto nz_cnt = indices.size(); in UnSparseTensorData() local
74 MS_ASSERT(nz_cnt == coors.size()); in UnSparseTensorData()
79 for (size_t i = 0; i < nz_cnt; i++) { in UnSparseTensorData()
|