Searched defs:src_batch (Results 1 – 8 of 8) sorted by relevance
28 const float *src_batch = src + i * col * deep; in PackLstmWeightFp32ToFp16() local36 const float16_t *src_batch = src + i * col * deep; in PackLstmWeightFp16() local46 const float *src_batch = src + i * col; in PackLstmBiasFp32ToFp16() local64 const float16_t *src_batch = src + i * col; in PackLstmBiasFp16() local
211 const float16_t *src_batch = (const float16_t *)src + n * batch; in PackNHWCToNCHWFp16() local459 const float *src_batch = src + b * plane * channel; in PackNHWCFp32ToNHWC8Fp16() local503 const float16_t *src_batch = src + b * plane * c8_channel; in PackNHWC8Fp16ToNHWCFp32() local518 const float16_t *src_batch = src + b * plane * c8_channel; in PackNHWC8ToNHWCFp16() local
27 const float *src_batch = (const float *)src + n * batch; in PackNHWCToNCHWFp32() local
27 const float *src_batch = src + i * col * deep; in PackLstmWeight() local42 const float *src_batch = src + i * col; in PackLstmBias() local
397 const char *src_batch = (char *)(src) + b * src_batch_offset; in PackNC8HW8AlignedToNC8HW8NotAlignedFp32() local564 const float *src_batch = (const float *)src + n * batch; in PackNHWCToNCHWFp32() local
138 …const float *src_batch = weight_h_data + i * lstm_param_->hidden_size_ * lstm_param_->hidden_size_; in InitStateWeightBias() local
345 const uint16_t *src_batch = static_cast<const uint16_t *>(src) + n * batch; in PackNHWCToNCHWFp16() local
1163 const int8_t *src_batch = (const int8_t *)src + n * batch; in PackNHWCToNCHWInt8() local