Searched refs:leaf_vars (Results 1 – 7 of 7) sorted by relevance
/dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/ |
H A D | schedule_lang.cc | 44 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 D | schedule_dataflow_rewrite.cc | 505 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 D | schedule_lang.cc | 44 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 D | schedule_dataflow_rewrite.cc | 505 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 D | schedule_lang.cc | 47 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 D | schedule_dataflow_rewrite.cc | 461 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 D | bv_subtheory_algebraic.cpp | 734 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()
|