Searched refs:PickDims (Results 1 – 2 of 2) sorted by relevance
99 out_dim_sizes = sp.PickDims(reduction.group_by_dims); in SparseTensorReduceHelper()
251 std::vector<int64> PickDims(gtl::ArraySlice<int64> dim_indices) const { in PickDims() function