Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcuda_codegen.cpp338 const std::vector<ExprPtr>& thread_extents = metavars.gpu_thread_extents(); in AtomicAddFuser() local
793 auto& thread_extents = cuda_analysis_->gpu_thread_extents(); in mutate() local
987 auto thread_extents = metavar_rewriter_->gpu_thread_extents(); in Initialize() local