Searched defs:loop_num (Results 1 – 6 of 6) sorted by relevance
29 int loop_num = parameter->loop_num_; in TopkInt8() local
43 int loop_num = parameter->loop_num_; in Topk() local
40 int loop_num = (row_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in GammaAndBetaThreadReduce() local113 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in InputThreadReduce() local137 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in InputThreadReduce() local
46 int loop_num = (row_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in GammaAndBetaThreadReduce() local129 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in InputThreadReduceInnerMean() local196 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in InputThreadReduceOuterMean() local235 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in InputThreadReduceOuterMean() local
51 int loop_num = (col_dim + NUM_PER_THREAD_REDUCE - 1) / NUM_PER_THREAD_REDUCE; in ThreadReduce() local
326 void KernelGraph::GetLoopNodesByDFS(const AnfNodePtr &node, uint32_t *loop_num) { in GetLoopNodesByDFS()373 uint32_t loop_num = 0; in GetLoopNum() local