Searched defs:thread_i (Results 1 – 5 of 5) sorted by relevance
124 State::State(size_t max_iters, const std::vector<int64_t>& ranges, int thread_i, in State()
161 const std::vector<int64_t>& ranges, int thread_i, int n_threads, in State()
169 const std::vector<int64_t>& ranges, int thread_i, int n_threads, in State()
959 const int thread_i = thread_id() / threads_per_row; in attention_kernel() local
1716 const int thread_i = thread_id / threads_per_row; in processBlockIJ() local