Home
last modified time | relevance | path

Searched defs:acc_buf (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dreduction.cpp34 BufHandle acc_buf, in operator ()()
59 BufHandle acc_buf, in make()
H A Dreduction.h158 BufPtr acc_buf, in ReduceOp()
197 void setAccBuf(BufHandle acc_buf) { in setAccBuf()
H A Dtensor.cpp19 auto acc_buf = reduce_ndim > 0 ? reduce_op->getAccBuf() : nullptr; in constructStmt() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh301 void* acc_buf; member
363 void* acc_buf; member