Searched refs:dst_data_size (Results 1 – 2 of 2) sorted by relevance
42 size_t dst_data_size) { in UnIndexTensorData() argument56 if (un_indexed_data.size() * sizeof(T) != dst_data_size) { in UnIndexTensorData()58 << " expected by tensor: " << dst_data_size; in UnIndexTensorData()70 … size_t elem_cnt, size_t coor_best_bit, void *dst_data, size_t dst_data_size) { in UnSparseTensorData() argument101 if (un_sparsed_data.size() * sizeof(T) > dst_data_size) { in UnSparseTensorData()103 << " tensor size: " << dst_data_size; in UnSparseTensorData()105 } else if (un_sparsed_data.size() * sizeof(T) < dst_data_size && in UnSparseTensorData()106 (un_sparsed_data.size() + (1 << coor_best_bit) - 1) * sizeof(T) < dst_data_size) { in UnSparseTensorData()107 …"un-sparsed data size: " << un_sparsed_data.size() * sizeof(T) << " tensor size: " << dst_data_size in UnSparseTensorData()112 for (; data_index < dst_data_size / sizeof(T); data_index++) { in UnSparseTensorData()
133 auto dst_data_size = weight_size; in AssignNewModelMemory() local134 int ret = memcpy_s(weight_data.get(), dst_data_size, weight.second->addr, src_data_size); in AssignNewModelMemory()