Home
last modified time | relevance | path

Searched refs:leaf_vars (Results 1 – 7 of 7) sorted by relevance

/dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/
H A Dschedule_lang.cc44 size_t pos = FindNodeRef(leaf_vars, v); in FindLeafVar()
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()
182 FindLeafVar(all_vars, leaf_vars, ivar); in bind()
213 leaf_vars->data.insert( in env_threads()
271 leaf_vars->data.erase(leaf_vars->data.begin() + pos_outer, in fuse()
273 leaf_vars->data.insert(leaf_vars->data.begin() + pos_outer, in fuse()
297 leaf_vars->data.insert(leaf_vars->data.begin(), singleton); in fuse()
330 leaf_vars->data[pos[i]] = temp[i]; in reorder()
[all …]
H A Dschedule_dataflow_rewrite.cc505 ArrayNode* leaf_vars = s->leaf_iter_vars.CopyOnWrite(); in RebaseNonZeroMinLoop() local
507 size_t idx = FindNodeRef(leaf_vars, iv); in RebaseNonZeroMinLoop()
514 if (idx < leaf_vars->data.size()) { in RebaseNonZeroMinLoop()
522 leaf_vars->data[idx] = rebased; in RebaseNonZeroMinLoop()
686 ArrayNode* leaf_vars = reduce_stage->leaf_iter_vars.CopyOnWrite(); in rfactor() local
688 size_t axis_pos = FindNodeRef(leaf_vars, axis); in rfactor()
689 CHECK_NE(axis_pos, leaf_vars->data.size()) in rfactor()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/schedule/
H A Dschedule_lang.cc44 size_t pos = FindNodeRef(leaf_vars, v); in FindLeafVar()
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()
182 FindLeafVar(all_vars, leaf_vars, ivar); in bind()
213 leaf_vars->data.insert( in env_threads()
271 leaf_vars->data.erase(leaf_vars->data.begin() + pos_outer, in fuse()
273 leaf_vars->data.insert(leaf_vars->data.begin() + pos_outer, in fuse()
297 leaf_vars->data.insert(leaf_vars->data.begin(), singleton); in fuse()
330 leaf_vars->data[pos[i]] = temp[i]; in reorder()
[all …]
H A Dschedule_dataflow_rewrite.cc505 ArrayNode* leaf_vars = s->leaf_iter_vars.CopyOnWrite(); in RebaseNonZeroMinLoop() local
507 size_t idx = FindNodeRef(leaf_vars, iv); in RebaseNonZeroMinLoop()
514 if (idx < leaf_vars->data.size()) { in RebaseNonZeroMinLoop()
522 leaf_vars->data[idx] = rebased; in RebaseNonZeroMinLoop()
686 ArrayNode* leaf_vars = reduce_stage->leaf_iter_vars.CopyOnWrite(); in rfactor() local
688 size_t axis_pos = FindNodeRef(leaf_vars, axis); in rfactor()
689 CHECK_NE(axis_pos, leaf_vars->data.size()) in rfactor()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/
H A Dschedule_lang.cc47 size_t pos = FindNodeRef(leaf_vars, v); in FindLeafVar()
48 if (pos < leaf_vars->size()) return pos; in FindLeafVar()
87 leaf_vars.erase(leaf_vars.begin() + pos); in SplitHelper()
88 leaf_vars.insert(leaf_vars.begin() + pos, inner); in SplitHelper()
89 leaf_vars.insert(leaf_vars.begin() + pos, outer); in SplitHelper()
177 FindLeafVar(all_vars, leaf_vars, ivar); in bind()
205 leaf_vars.insert(leaf_vars.begin(), temp.begin(), temp.end()); in env_threads()
258 leaf_vars.erase(leaf_vars.begin() + pos_outer, leaf_vars.begin() + pos_inner + 1); in fuse()
259 leaf_vars.insert(leaf_vars.begin() + pos_outer, fused); in fuse()
281 leaf_vars.insert(leaf_vars.begin(), singleton); in fuse()
[all …]
H A Dschedule_dataflow_rewrite.cc461 ArrayNode* leaf_vars = s->leaf_iter_vars.CopyOnWrite(); in RebaseNonZeroMinLoop() local
463 size_t idx = FindNodeRef(leaf_vars, iv); in RebaseNonZeroMinLoop()
469 if (idx < leaf_vars->size()) { in RebaseNonZeroMinLoop()
476 leaf_vars->SetItem(idx, rebased); in RebaseNonZeroMinLoop()
708 ArrayNode* leaf_vars = reduce_stage->leaf_iter_vars.CopyOnWrite(); in rfactor() local
710 size_t axis_pos = FindNodeRef(leaf_vars, axis); in rfactor()
711 CHECK_NE(axis_pos, leaf_vars->size()) << "Cannot find IterVar " << axis << " in leaf iter vars"; in rfactor()
/dports/math/cvc4/CVC4-1.7/src/theory/bv/
H A Dbv_subtheory_algebraic.cpp734 NodeSet leaf_vars; in collectModelInfo() local
741 collectVariables(subst, leaf_vars); in collectModelInfo()
746 for (NodeSet::const_iterator it = leaf_vars.begin(); it != leaf_vars.end(); ++it) { in collectModelInfo()