Home
last modified time | relevance | path

Searched defs:reduce_args (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Dreduction.h112 const std::vector<VarPtr>& reduce_args) { in complete()
123 const std::vector<VarHandle>& reduce_args) { in complete()
143 std::vector<VarPtr> reduce_args, in ReduceOp()
156 std::vector<VarPtr> reduce_args, in ReduceOp()
193 const std::vector<VarPtr>& reduce_args() const { in reduce_args() function
Dreduction.cpp49 const std::vector<VarHandle>& reduce_args, in make()
57 const std::vector<VarHandle>& reduce_args, in make()
Dtensor.h26 const std::vector<VarPtr>& reduce_args, in Tensor()
35 const std::vector<VarHandle>& reduce_args, in Tensor()
Dtensorexpr_init.cpp352 const std::vector<ExprHandle>& reduce_args) { in initTensorExprBindings()
363 const std::vector<ExprHandle>& reduce_args) { in initTensorExprBindings()
374 const std::vector<ExprHandle>& reduce_args) { in initTensorExprBindings()
387 const std::vector<ExprHandle>& reduce_args) { in initTensorExprBindings()
Dloopnest.cpp2906 std::set<VarPtr> reduce_args( in cacheAccesses() local
3298 std::set<VarPtr> reduce_args = { in rfactor() local