Lines Matching refs:leaf_vars

43 size_t FindLeafVar(ArrayNode* all_vars, ArrayNode* leaf_vars, const IterVar& v) {  in FindLeafVar()  argument
44 size_t pos = FindNodeRef(leaf_vars, v); in FindLeafVar()
45 if (pos < leaf_vars->data.size()) return pos; in FindLeafVar()
76 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in Split() local
77 size_t pos = FindLeafVar(all_vars, leaf_vars, parent); in Split()
83 leaf_vars->data.erase(leaf_vars->data.begin() + pos); in Split()
84 leaf_vars->data.insert(leaf_vars->data.begin() + pos, inner); in Split()
85 leaf_vars->data.insert(leaf_vars->data.begin() + pos, outer); in Split()
181 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in bind() local
182 FindLeafVar(all_vars, leaf_vars, ivar); in bind()
207 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in env_threads() local
213 leaf_vars->data.insert( in env_threads()
214 leaf_vars->data.begin(), temp.begin(), temp.end()); in env_threads()
259 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in fuse() local
261 size_t pos_inner = FindLeafVar(all_vars, leaf_vars, inner); in fuse()
262 size_t pos_outer = FindLeafVar(all_vars, leaf_vars, outer); in fuse()
271 leaf_vars->data.erase(leaf_vars->data.begin() + pos_outer, in fuse()
272 leaf_vars->data.begin() + pos_inner + 1); in fuse()
273 leaf_vars->data.insert(leaf_vars->data.begin() + pos_outer, in fuse()
295 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in fuse() local
297 leaf_vars->data.insert(leaf_vars->data.begin(), singleton); in fuse()
318 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in reorder() local
322 pos.push_back(FindLeafVar(all_vars, leaf_vars, order[i])); in reorder()
326 temp.emplace_back(leaf_vars->data[pos[i]]); in reorder()
330 leaf_vars->data[pos[i]] = temp[i]; in reorder()
352 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in UpdateIterVarAttr() local
353 FindLeafVar(all_vars, leaf_vars, var); in UpdateIterVarAttr()
422 ArrayNode* leaf_vars = self->leaf_iter_vars.CopyOnWrite(); in prefetch() local
423 FindLeafVar(all_vars, leaf_vars, var); in prefetch()