Searched defs:BufferArg (Results 1 – 2 of 2) sorted by relevance
125 BufferArg(const Tensor& tensor) : buf_(tensor.buf()) {} in BufferArg() function126 BufferArg(const VarHandle& var) : var_(var.node()), isVar_(true) {} in BufferArg() function127 BufferArg(const BufHandle& buf) : buf_(buf.node()) {} in BufferArg() function128 BufferArg(BufPtr buf) : buf_(std::move(buf)) {} in BufferArg() function
83 using BufferArg = CodeGen::BufferArg; member