Home
last modified time | relevance | path

Searched refs:fused_iter (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/search_policy/
H A Dutils.h572 inline State FuseAllReductionIterators(const State& state, int stage_id, Iterator* fused_iter, in FuseAllReductionIterators() argument
589 *fused_iter = tmp_s.fuse(stage_id, *reduce_iters); in FuseAllReductionIterators()
591 *fused_iter = (*reduce_iters)[0]; in FuseAllReductionIterators()
597 inline State FuseAllOuterSpaceIterators(const State& state, int stage_id, Iterator* fused_iter) { in FuseAllOuterSpaceIterators() argument
615 *fused_iter = tmp_s.fuse(stage_id, to_fuse); in FuseAllOuterSpaceIterators()
617 *fused_iter = to_fuse[0]; in FuseAllOuterSpaceIterators()
H A Dsketch_policy_rules.cc572 Iterator fused_iter = state->fuse(stage_id, to_fuse); in Apply() local
573 state->parallel(stage_id, fused_iter); in Apply()