Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DReduce.cuh1165 int64_t output_memory_size = iter.element_size(0); in gpu_reduce_kernel() local
1273 int64_t output_memory_size = iter.element_size(0); in jitted_gpu_reduce_kernel() local