Searched defs:src_batch (Results 1 – 11 of 11) sorted by relevance
87 …const float *src_batch = weight_h_data + i * lstm_param_->hidden_size_ * lstm_param_->output_size_; in InitStateWeightBias() local142 …const float *src_batch = weight_pro_data + i * lstm_param_->hidden_size_ * lstm_param_->output_siz… in InitProjectWeight() local
135 …const float *src_batch = weight_h_data + i * lstm_param_->hidden_size_ * lstm_param_->output_size_; in InitStateWeightBias() local195 …const float *src_batch = weight_proj_data + i * lstm_param_->hidden_size_ * lstm_param_->output_si… in InitProjectWeight() local
25 static void PackLstmMatrix(const float *src_batch, float *dst_batch, int col, int deep) { in PackLstmMatrix()38 const float *src_batch = src + i * col * deep; in PackLstmWeightBatch() local63 const float *src_batch = src + i * col; in PackLstmBias() local82 const float *src_batch = src + i * col; in PackLstmBiasWithStride() local
1480 const char *src_batch = (char *)(src) + b * src_batch_offset; in PackNC8HW8AlignedToNC8HW8NotAlignedFp32() local1647 const float *src_batch = (const float *)src + n * batch; in PackNHWCToNCHWFp32() local1719 const float *src_batch = (const float *)src + b * b_stride; in TransposeFp32() local
29 const float *src_batch = src + i * col * deep; in PackLstmWeightFp32ToFp16() local42 const float16_t *src_batch = src + i * col * deep; in PackLstmWeightFp16() local56 const float *src_batch = src + i * col; in PackLstmBiasFp32ToFp16() local75 const float16_t *src_batch = src + i * col; in PackLstmBiasFp16() local
139 const float16_t *src_batch = (const float16_t *)src + n * batch; in PackNHWCToNCHWFp16() local394 const float *src_batch = src + b * plane * channel; in PackNHWCFp32ToNHWC8Fp16() local462 const float16_t *src_batch = src + b * plane * c8_channel; in PackNHWC8Fp16ToNHWCFp32() local477 const float16_t *src_batch = src + b * plane * c8_channel; in PackNHWC8ToNHWCFp16() local
24 const float *src_batch = (const float *)src + n * batch; in PackNHWCToNCHWFp32() local
48 const float *src_batch = src + i * col * row; in PackLstmWeightTranspose() local
317 const int8_t *src_batch = (const int8_t *)src + n * batch; in PackNHWCToNCHWInt8() local
640 const uint16_t *src_batch = static_cast<const uint16_t *>(src) + n * batch; in PackNHWCToNCHWFp16() local
324 const VkSubmitInfo *src_batch = &submit->submit_batches[batch_index]; in vn_fix_device_group_cmd_count() local