Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/tensorexpr/
Dir_mutator.cpp320 std::vector<VarPtr> new_reduce_args; in mutate() local
322 new_reduce_args.push_back(static_to<Var>(r->accept_mutator(this))); in mutate()
325 return alloc<ReduceOp>(body_new, new_reduce_args, v->reducer()); in mutate()
Dloopnest.cpp3235 std::vector<VarPtr> new_reduce_args; in mutate() local
3238 new_reduce_args.push_back(r); in mutate()
3242 return alloc<ReduceOp>(body_new, new_reduce_args, v->reducer()); in mutate()