Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/sparse/
DSparseBinaryOpIntersectionCommon.h96 auto idx_sizes = std::vector<int64_t>(res_values.dim(), 1); in make_value_selection_intersection_iter() local
DValidateCompressedIndicesCommon.h298 const auto idx_sizes = std::get<0>(*idx_geometry_holder); in _validate_compressed_sparse_indices_kernel() local