Searched defs:output_memory_size (Results 1 – 1 of 1) sorted by relevance
1165 int64_t output_memory_size = iter.element_size(0); in gpu_reduce_kernel() local1273 int64_t output_memory_size = iter.element_size(0); in jitted_gpu_reduce_kernel() local