Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/tensorexpr/
Dcuda_codegen.cpp326 const std::vector<ExprPtr>& block_extents = metavars.gpu_block_extents(); in AtomicAddFuser() local
807 auto& block_extents = cuda_analysis_->gpu_block_extents(); in mutate() local
986 auto block_extents = metavar_rewriter_->gpu_block_extents(); in Initialize() local