Searched refs:weights_ptr (Results 1 – 3 of 3) sorted by relevance
2051 const int* weights_ptr = in ConvertTranspose() local2053 std::vector<int> perm(weights_ptr, weights_ptr + weights.count()); in ConvertTranspose()2092 const int* weights_ptr = in ConvertReshape() local2128 const int reshape_batch_dim = weights_ptr[0]; in ConvertReshape()2134 reshape_dims.d[i - 1] = weights_ptr[i]; in ConvertReshape()2197 const int* weights_ptr = in ConvertExpandDims() local2199 int axis = weights_ptr[0]; in ConvertExpandDims()
331 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local332 uint8x16_t weights_val_u8_0 = vld1q_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()333 uint8x16_t weights_val_u8_1 = vld1q_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()334 uint8x16_t weights_val_u8_2 = vld1q_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()335 uint8x16_t weights_val_u8_3 = vld1q_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()407 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local408 uint8x8_t weights_val_u8_0 = vld1_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()409 uint8x8_t weights_val_u8_1 = vld1_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()410 uint8x8_t weights_val_u8_2 = vld1_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()411 uint8x8_t weights_val_u8_3 = vld1_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()
453 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local454 uint8x16_t weights_val_u8_0 = vld1q_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()455 uint8x16_t weights_val_u8_1 = vld1q_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()456 uint8x16_t weights_val_u8_2 = vld1q_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()457 uint8x16_t weights_val_u8_3 = vld1q_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()529 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local530 uint8x8_t weights_val_u8_0 = vld1_u8(weights_ptr + 0 * input_size); in GEMVForLstmCell()531 uint8x8_t weights_val_u8_1 = vld1_u8(weights_ptr + 1 * input_size); in GEMVForLstmCell()532 uint8x8_t weights_val_u8_2 = vld1_u8(weights_ptr + 2 * input_size); in GEMVForLstmCell()533 uint8x8_t weights_val_u8_3 = vld1_u8(weights_ptr + 3 * input_size); in GEMVForLstmCell()