Searched refs:weights_ptr (Results 1 – 3 of 3) sorted by relevance
194 const float* weights_ptr = GetTensorData<float>(weights); in Eval() local242 const float w = weights_ptr[i]; in Eval()
1192 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local1193 uint8x16_t weights_val_u8_0 = vld1q_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()1194 uint8x16_t weights_val_u8_1 = vld1q_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()1195 uint8x16_t weights_val_u8_2 = vld1q_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()1196 uint8x16_t weights_val_u8_3 = vld1q_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()1268 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local1269 uint8x8_t weights_val_u8_0 = vld1_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()1270 uint8x8_t weights_val_u8_1 = vld1_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()1271 uint8x8_t weights_val_u8_2 = vld1_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()1272 uint8x8_t weights_val_u8_3 = vld1_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()
2498 const int* weights_ptr = static_cast<int*>(weights.GetValues()); in ConvertTranspose() local2499 std::vector<int> perm(weights_ptr, weights_ptr + weights.count()); in ConvertTranspose()6101 const int* weights_ptr = static_cast<int*>(weights.GetValues()); in ConvertResize() local6149 output_dimensions.d[output_dimensions.nbDims - 2] = weights_ptr[0]; in ConvertResize()6150 output_dimensions.d[output_dimensions.nbDims - 1] = weights_ptr[1]; in ConvertResize()