Searched defs:dst_batch (Results 1 – 10 of 10) sorted by relevance
88 … float *dst_batch = weight_h_ptr_ + i * lstm_param_->state_col_align_ * lstm_param_->output_size_; in InitStateWeightBias() local143 float *dst_batch = weight_project_ptr_ + i * lstm_param_->hidden_size_ * col_align; in InitProjectWeight() local
136 float *dst_batch = in InitStateWeightBias() local196 …float *dst_batch = weight_project_ptr_ + i * lstm_param_->hidden_size_ * lstm_param_->proj_col_ali… in InitProjectWeight() local
25 static void PackLstmMatrix(const float *src_batch, float *dst_batch, int col, int deep) { in PackLstmMatrix()39 float *dst_batch = dst + ((order == NULL) ? i : order[i]) * col_align * deep; in PackLstmWeightBatch() local64 float *dst_batch = dst + ((order == NULL) ? i : order[i]) * col_align; in PackLstmBias() local83 float *dst_batch = dst + ((order == NULL) ? i : order[i]) * col_align; in PackLstmBiasWithStride() local
1648 float *dst_batch = (float *)dst + n * batch; in PackNHWCToNCHWFp32() local1720 float *dst_batch = (float *)dst + b * b_stride; in TransposeFp32() local
30 float16_t *dst_batch = dst + (order == NULL ? i : order[i]) * col_align * deep; in PackLstmWeightFp32ToFp16() local43 float16_t *dst_batch = dst + (order == NULL ? i : order[i]) * col_align * deep; in PackLstmWeightFp16() local57 float16_t *dst_batch = dst + (order == NULL ? i : order[i]) * col_align; in PackLstmBiasFp32ToFp16() local76 float16_t *dst_batch = dst + (order == NULL ? i : order[i]) * col_align; in PackLstmBiasFp16() local
140 float16_t *dst_batch = (float16_t *)dst + n * batch; in PackNHWCToNCHWFp16() local393 float16_t *dst_batch = dst + b * plane * c8_channel; in PackNHWCFp32ToNHWC8Fp16() local463 float *dst_batch = dst + b * plane * channel; in PackNHWC8Fp16ToNHWCFp32() local478 float16_t *dst_batch = dst + b * plane * channel; in PackNHWC8ToNHWCFp16() local
25 float *dst_batch = reinterpret_cast<float *>(dst) + n * batch; in PackNHWCToNCHWFp32() local
49 float *dst_batch = dst + ((order == NULL) ? i : order[i]) * col * row_align; in PackLstmWeightTranspose() local
318 int8_t *dst_batch = (int8_t *)dst + n * batch; in PackNHWCToNCHWInt8() local
641 uint16_t *dst_batch = static_cast<uint16_t *>(dst) + n * batch; in PackNHWCToNCHWFp16() local