Home
last modified time | relevance | path

Searched refs:intersection_buffer (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/
DSparseBinaryOpIntersectionCommon.h328 auto intersection_buffer = at::empty({2, source_nnz}, sorted_hash.options()); variable
329 auto intersection_count = intersection_buffer.select(0, 0);
330 auto intersection_first_idx = intersection_buffer.select(0, 1);