Ah nvm, snpe just can't do 8x8 matmuls

stop-lines
mitchellgoffpc 2021-09-16 12:11:49 -07:00
parent b286f41f75
commit 53b013b1fb
1 changed files with 1 additions and 1 deletions

View File

@ -45,7 +45,7 @@ constexpr int LEAD_IDX = RE_IDX + 2*2*2*33;
constexpr int LEAD_PROB_IDX = LEAD_IDX + LEAD_MHP_N*(LEAD_MHP_GROUP_SIZE);
constexpr int STOP_LINE_IDX = LEAD_PROB_IDX + 3;
constexpr int STOP_LINE_PROB_IDX = STOP_LINE_IDX + STOP_LINE_MHP_N*STOP_LINE_MHP_GROUP_SIZE;
constexpr int DESIRE_STATE_IDX = STOP_LINE_PROB_IDX + 2;
constexpr int DESIRE_STATE_IDX = STOP_LINE_PROB_IDX + 1;
constexpr int META_IDX = DESIRE_STATE_IDX + DESIRE_LEN;
constexpr int POSE_IDX = META_IDX + OTHER_META_SIZE + DESIRE_PRED_SIZE;
constexpr int OUTPUT_SIZE = POSE_IDX + POSE_SIZE;