Searched defs:thread_local_bufs_ (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/ | ||
| D | cuda_codegen.h | 60 std::unordered_set<VarPtr> thread_local_bufs_; variable |
| D | cuda_codegen.cpp | 401 const std::unordered_set<VarPtr>& thread_local_bufs_; member in torch::jit::tensorexpr::AtomicAddFuser |