Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcuda_codegen.h45 const std::vector<ExprPtr>& gpu_thread_extents() const { in gpu_thread_extents() function
107 const std::vector<ExprPtr>& gpu_thread_extents() const { in gpu_thread_extents() function
240 const std::vector<ExprPtr>& gpu_thread_extents() const { in gpu_thread_extents() function
H A Dcuda_codegen.cpp1076 size_t gpu_thread_extents = thread_block_size_; in call_with_numel() local
1121 const std::vector<ExprPtr>& gpu_thread_extents = in call_raw() local