Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp900 const auto sparse_dim_intersec = std::min(sparse_dim, d_dim - dense_dim); in intersection_binary_op_sparse_dense_out() local
917 for (const auto i : c10::irange(sparse_dim_intersec)) { in intersection_binary_op_sparse_dense_out()
921 for (auto i = d_start_dim_intersec + sparse_dim_intersec; i < d_dim; ++i) { in intersection_binary_op_sparse_dense_out()