Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A 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
H A Dreduction.cpp49 const std::vector<VarHandle>& reduce_args, in make()
57 const std::vector<VarHandle>& reduce_args, in make()
H A Dtensor.h26 const std::vector<VarPtr>& reduce_args, in Tensor()
35 const std::vector<VarHandle>& reduce_args, in Tensor()
H A 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()
H A Dloopnest.cpp2906 std::set<VarPtr> reduce_args( in cacheAccesses() local
3298 std::set<VarPtr> reduce_args = { in rfactor() local