Searched defs:row_indices (Results 1 – 2 of 2) sorted by relevance
20 …indexing_functor(const ArgType& arg, const RowIndexType& row_indices, const ColIndexType& col_indi… in indexing_functor()33 mat_indexing(const Eigen::MatrixBase<ArgType>& arg, const RowIndexType& row_indices, const ColIndex… in mat_indexing()
109 GpuLaunchConfig cfg, int rank, const Tindex* indices, Tindex* row_indices) { in CopyRowIndicesKernel()463 auto row_indices = row_indices_t.flat<Tindex>(); in ArgSortByRows() local